huggingface / huggingface/evaluate

feature request: Dice metric for segmentation models

Open
#414 3 comments 0 reactions 0 assignees View on GitHub
Dominant language
Python
Stars
2.5k
Forks
341
PR merge metrics
No merged PRs in 30d

Description

The library already has [mean_iou (mean Jaccard index)](https://en.wikipedia.org/wiki/Jaccard_index) which I have used for my image segmentation models and it works well.

If you look at recent research papers on the topic of image segmentation for medical imaging, you will notice that [the Dice metric (Sorensen-Dice coefficient)](https://en.wikipedia.org/wiki/S%C3%B8rensen%E2%80%93Dice_coefficient) is used a lot in research.

It would be very useful to have the Dice metric integrated into the main library - that would simplify model comparison when reviewing the literature. Thanks!

Contributor guide

Open the contributing guide

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.