深度学习pytorch-unet训练报错f’Either no mask or multiple masks found for the ID {idx}: {mask_file}’
做了如下改动:
1.查看数据集后缀,我的之前是.tif后缀,更改后是.jpg。
2.在项目dataset.py文件的箭头指向处加“.”。
解决了。
[已解决]深度学习中unet训练报错f‘Either no mask or multiple masks found for the ID {idx}: {mask_file}‘
最新推荐文章于 2024-08-14 18:54:06 发布
"博主分享了如何通过修改数据集后缀从.tif转换为.jpg解决深度学习中使用PyTorch UNet训练时遇到的错误,即'Either no mask or multiple masks found for the ID {idx}
摘要由CSDN通过智能技术生成