Replace `torchmetrics.classifcation.Dice` with `torchmetrics.segmentation.GeneralizedDice`

Sllambias / yucca

Apache License 2.0

14 stars 1 forks source link

Replace `torchmetrics.classifcation.Dice` with `torchmetrics.segmentation.GeneralizedDice` #191

Closed asbjrnmunk closed 3 days ago

asbjrnmunk commented 1 week ago

torchmetrics.classifcation.Dice is just broken, but GeneralizedDice seems to work well. This also means we can get rid of F1 score and get dice for each label directly.