Retina U-Net does not work in the toy experiment
skief opened this issue · comments
Simon commented
When I try to use the Retina U-Net in the toy experiment(just changed self.model = 'retina_unet' and the path to the data) the program crashes in line 447 of the retina_unet.py due to the var_seg[:, 0]
containing the values [0, 1, 2]
while the size of the seg_logits
is just [20, 2, 320, 320]
and not [20, 3, 320, 320]
Gregor1337 commented
Hi skief,
thanks for your input.
Fixed in this commit.