Pytorch-UNet
Pytorch-UNet copied to clipboard
AssertionError: True mask indices should be in [0, 1]
但是输出的结果也是在合理范围内呀
有大佬帮忙看看什么原因吗。谢谢啦
但是输出的结果也是在合理范围内呀
有大佬帮忙看看什么原因吗。谢谢啦
hi 问题解决了吗?运行时会分析掩膜文件,输出掩膜值,可以看一下输出的结果是什么,比如我这边的输出是这样的:
INFO: Scanning mask files to determine unique values
100%|███████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 22806/22806 [00:15<00:00, 1494.46it/s]
INFO: Unique mask values: [0, 1, 2, 3, 255]
因为只有一个分类,所以我想这个unique mask values的值过多了,调试打印出来的mask_true.min(),mask_true.max()分别是tensor(0, device='cuda:0') tensor(4, device='cuda:0')