Segmentation UNET results

[Meddebma]

Hi,
I was implementing the UNET segmentation tutorial and the training went well, but when I tried to display the segmentation results, the output was completely empty, is it a problem of training or a matter of display?
here is the code
https://github.com/Meddebma/pyradiomics/blob/master/Untitled0.ipynb

thank you

[Nic-Ma]

Hi @yiheng-wang-nv ,

Could you please help check the notebook?

Thanks in advance.

[Nic-Ma]

Hi @Meddebma ,

I think this ticket is duplicated, can we close it and discuss in #2668 directly？

Thanks.

[Meddebma]

Hi @Nic-Ma,
as you like, the difference is that in this notebook I used the UNET tutorial with only one output_channel, in the other Ticket I am using the two channels output in the model architecture (Which works well) and I am trying to remove the background channel