Amshaker / unetr_plus_plus

[IEEE TMI-2024] UNETR++: Delving into Efficient and Accurate 3D Medical Image Segmentation
Apache License 2.0
340 stars 32 forks source link

How to get the Haussdorff Distance? #43

Closed Leonngm closed 1 year ago

Leonngm commented 1 year ago

Hi all,

I wonder how I can evaluate HD95. When running the evaluation scrip, only the dice score is calculated. Thanks.

Best regards

Amshaker commented 1 year ago

Hi @Leonngm ,

Please refer to this function for HD evaluation. You need to run inference_DATASET file to get the HD95 metric. https://github.com/Amshaker/unetr_plus_plus/blob/bc6bed1eb2d2d735d5f44740958ef4ea862304a6/unetr_pp/inference_synapse.py#L16