[Deep Learning] [Original] pytorch-unet training error Either no mask or multiple masks found for the ID

Project address: https://github.com/milesial/Pytorch-UNet

Use custom data to report errors

Error content:

Either no mask or multiple masks found for the ID {idx}: {mask_file}'
AssertionError: Either no mask or multiple masks found for the ID

In fact, this is a very simple solution.

dir_img = 'data/mydata/train/pic/'
dir_mask = 'data/mydata/train/cv2_mask/'

You must add a backslash to your own path at the end, otherwise the above error will be reported when you get the picture.