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

self.label? #13

Open pakheiyeung opened 3 years ago

pakheiyeung commented 3 years ago

https://github.com/MIC-DKFZ/basic_unet_example/blob/45dd8b8e6767448ff859203c834d01bd94f68ab9/datasets/three_dim/NumpyDataLoader.py#L164 Should the self.input be replaced by 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!