About output & label

[AIddddddd]

I'm interested in your research, but when running the code, there is a problem, my output and label shape do not match,output.shape = (8, 1, 352, 352)
label.shape = (8, 1, 352, 352, 3) please why is this, thanks

[AIddddddd]

Sorry，this problem occurred https://github.com/jcwang123/BA-Transformer ，it is not this program，Looking forward to your reply

[jcwang123]

Maybe the channel (...,3) is the RGB channel while our label should be single-channel. You can select any one channel as the final label, like:
label = label[...,0]

[AIddddddd]

thank you，i got it

[AIddddddd]

Maybe the channel (...,3) is the RGB channel while our label should be single-channel. You can select any one channel as the final label, like: label = label[...,0]

Thanks for your answer, I have another question, I am running the isic2016-part1-test dataset on the CPU, but his results are not good, is this due to the hardware gap?

Dice value of test dataset : 0.736586
Jc value of test dataset : 0.641301
Hd95 value of test dataset : 25.044101
Assd value of test dataset : 4.535037