Pytorch-UNet The predicted result graph cannot be visualized

The predicted result graph cannot be visualized

Open heixinbaicaishang opened this issue 3 years ago • 1 comments

Thank you very much for the UNET code, but the predicted output of the picture is 0, as shown in the figure below. How to solve this problem?

Dec 05 '21 14:12 heixinbaicaishang

ax[i + 1].imshow(mask[:, :, i]) Should be ax[i + 1].imshow(mask[1, :, :]) Problem solving！！！

Dec 14 '21 10:12 heixinbaicaishang