self.label?

[pakheiyeung]

basic_unet_example/datasets/three_dim/NumpyDataLoader.py

Line 164 in 45dd8b8

labels.append(numpy_array[list(self.input)]) # 'None' keeps the dimension

Should the self.input be replaced by 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!