Problems in the UNetExperiment3D

[RivettiLuciano]

Hi Guys, there is an issue when loading the datasets on the setup function. The segmentation in the datasets are initialized as the current images you want to predict/train. To solve this, you could change the line 164 of the file /basic_unet_example/datasets/three_dim/NumpyDataLoader.py for labels.append(numpy_array[list(self.label)]).

[elpequeno]

Sorry for the late reply. I was not looking into this repo for quite some while. I think you are right, I will have a look and correct it. Thank you!