Pytorch-UNet icon indicating copy to clipboard operation
Pytorch-UNet copied to clipboard

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

Open zhengys66 opened this issue 1 year ago • 1 comments

b1c1d8350e37f59ada7df365bd0f4e8

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

e45ecf67939d6cf13f3256fc7db65c0

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

zhengys66 avatar Aug 08 '23 03:08 zhengys66

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

danny-zhu avatar Jan 02 '24 02:01 danny-zhu