lee-zq/3DUNet-Pytorch

Dice loss problem

Opened this issue · 0 comments

Hi, thank you for your amazing work.
I thing there is some problem for dice loss, especially when u calculate with weight class.
Regrading this paper:
https://www.ncbi.nlm.nih.gov/pmc/articles/PMC8393549/

the weight should be apply to both numerator and de numerator.

I have try your dice loss code and loss give negative value. I thing this problem cause by the values of numerator is too high.