willianLee

train got error: The above exception was the direct cause of the following exception: Traceback (most recent call last): File "/media/lee/common/PycahrmProjects/Pytorch-UNet-master/train.py", line 43, in train_model dataset = CarvanaDataset(dir_img, dir_mask, img_scale)...

> thanks for this. i got new problem, when i set model channel=3, it report error: AssertionError: Network has been defined with 3 input channels, but loaded images have 1...