milesial / Pytorch-UNet

PyTorch implementation of the U-Net for image semantic segmentation with high quality images
GNU General Public License v3.0
8.65k stars 2.4k forks source link

AssertionError: True mask indices should be in [0, 1] #453

Open zhengys66 opened 10 months ago

zhengys66 commented 10 months ago

b1c1d8350e37f59ada7df365bd0f4e8

但是输出的结果也是在合理范围内呀

e45ecf67939d6cf13f3256fc7db65c0

有大佬帮忙看看什么原因吗。谢谢啦

danny-zhu commented 6 months ago

b1c1d8350e37f59ada7df365bd0f4e8

但是输出的结果也是在合理范围内呀

e45ecf67939d6cf13f3256fc7db65c0

有大佬帮忙看看什么原因吗。谢谢啦

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')