search

cosmic-cortex / pytorch-UNet

2D and 3D UNet implementation in PyTorch.
MIT License
147 stars 40 forks source link

valueError #2

Open 15793723081 opened 4 years ago

15793723081 commented 4 years ago

image Hello, I have read your code, which is very helpful to me. However, I reported such an error when using my own data set. Is there any solution?Thanks a million!

cosmic-cortex commented 4 years ago

Judging from the ValueError, I think the problem is that the image to be saved is constant in intensity, thus it cannot convert it to uint8. I'll investigate this problem a bit and get back to you soon!

15793723081 commented 4 years ago

Thank you and look forward to your reply.