the code is using 0.5 as threshold for the forground prob. which is arbitrary. It should be done with argmax(). https://github.com/dl4mia/03_segmentation_unet/blob/c0223dc4c699f19fb2a4b2ae0de5bfd04ab3407a/PixelClassification/test.py#L47 the same exists for 3D version as well.
the code is using 0.5 as threshold for the forground prob. which is arbitrary. It should be done with argmax().
03_segmentation_unet/PixelClassification/test.py
Line 47 in c0223dc
the same exists for 3D version as well.