pytorch Unet训练

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

开始训练报错:Either no mask or multiple masks found for the ID 

问题出在BasicDataset中的mask_suffix,其默认值为空,检查自己的mask名字如果不是空则会报错

posted @ 2020-09-29 17:21  AI_ON  阅读(802)  评论(1)    收藏  举报