search

MIC-DKFZ / basic_unet_example

An example project of how to use a U-Net for segmentation on medical images with PyTorch.
Apache License 2.0
141 stars 38 forks source link

Problems in the UNetExperiment3D #15

Open RivettiLuciano opened 3 years ago

RivettiLuciano commented 3 years ago

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 commented 3 years ago

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!