Modifications for conventional image (1 or 3 channels)

[Scienceseb]

Hello, the code of the model is relatively complex and difficult to modify to consider images of conventional size either with 1 channel (black and white) or 3 channels (RGB). Could you advise me what to modify to have a variable number of channels. Thanks