train_unet_demo produces wrong image

[khlee0192]

I am working on applying different loss functions on unet module.

To do that, I tried to run train_unet_demo.py file with channel of 128
(the leaderboard code has 256, but I had to decrease channels due to GPU memory)

However, tested images comes out to be very strange. Providing my image and ground truth image, my image seems to have some kind of shape but not the exact type.

I tried to fix this by changing color map or manually calculating pixel values, but did not work well. Could you give any idea to solve this issue?

• GPU : 3060 Ti 8GB RAM
• I don't know exact source of dataset, since it is given by my supervisor.

[khlee0192]

Though I reported this issue as bug, I think it would be problem on my code.