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

IndexError: Target 255 is out of bounds.

Open mmmmyolo opened this issue 3 years ago • 4 comments

image 更换数据集后出现的问题

mmmmyolo avatar Dec 05 '22 11:12 mmmmyolo

Hi, can you try the latest master version? Custom dataset support was improved.

milesial avatar Dec 06 '22 19:12 milesial

Hi, can you try the latest master version? Custom dataset support was improved.

image Hi,I use the 3.0 version,I find the reason of this problem is that my dataset label is binary picture it only has 01 pixel,but I find the true_mask is 0 and 255,it cause the problem.I don't know the reason why the pixel will be changed in the network~

mmmmyolo avatar Dec 07 '22 02:12 mmmmyolo

Yes, please try the latest master version, this should be fixed

milesial avatar Dec 07 '22 11:12 milesial

Yes, please try the latest master version, this should be fixed

OK,thank you

mmmmyolo avatar Dec 07 '22 11:12 mmmmyolo