pytorch-unet训练报错Either nomask or multiple masks found for the ID

项目地址：https://github.com/milesial/Pytorch-UNet

报错内容：

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

AssertionError:Either no mask or multiple masks found for the ID.

 

解决方案：

data_loading中的CarvanaDataset类下的改为mask_suffix='', 附上修改后的代码：

class CarvanaDataset(BasicDataset):
    def __init__(self, images_dir, masks_dir, scale=1):
        super().__init__(images_dir, masks_dir, scale, mask_suffix=''

# mask_suffix='_mask'被修改为mask_suffix=''