4uiiurz1/pytorch-nested-unet

Please change the validation code.

MinchangSung0223 opened this issue · 1 comments

MinchangSung0223 commented

If the code is not modified, the same output file is created.

pytorch-nested-unet/val.py

Line 105 in 1424f72

(output[i, 0] * 255).astype('uint8'))

(output[i, c] * 255).astype('uint8'))

4uiiurz1 commented

@tjdalsckd Thanks for letting me know! I fixed it.