LeeJunHyun / Image_Segmentation

Pytorch implementation of U-Net, R2U-Net, Attention U-Net, and Attention R2U-Net.
2.66k stars 594 forks source link

Grayscale image #81

Closed nick0622 closed 2 years ago

nick0622 commented 2 years ago

Excuse me, if I want to run the model to train some grayscale images. Where should the codes be modified?

nick0622 commented 2 years ago

!python main.py --img_ch=1