From e4b9d72a11ae55a3900aacd38f36ef9472e0a003 Mon Sep 17 00:00:00 2001 From: MengzhangLI Date: Thu, 4 Aug 2022 18:32:52 +0800 Subject: [PATCH] [Fix] 2.0 Fix mDice metric for medical datasets (#1864) --- configs/_base_/datasets/chase_db1.py | 2 +- configs/_base_/datasets/drive.py | 2 +- configs/_base_/datasets/hrf.py | 2 +- configs/_base_/datasets/stare.py | 2 +- mmseg/datasets/chase_db1.py | 2 +- 5 files changed, 5 insertions(+), 5 deletions(-) diff --git a/configs/_base_/datasets/chase_db1.py b/configs/_base_/datasets/chase_db1.py index 1622bec95..8cd4f3c28 100644 --- a/configs/_base_/datasets/chase_db1.py +++ b/configs/_base_/datasets/chase_db1.py @@ -55,5 +55,5 @@ val_dataloader = dict( pipeline=test_pipeline)) test_dataloader = val_dataloader -val_evaluator = dict(type='IoUMetric', iou_metrics=['mIoU']) +val_evaluator = dict(type='IoUMetric', iou_metrics=['mDice']) test_evaluator = val_evaluator diff --git a/configs/_base_/datasets/drive.py b/configs/_base_/datasets/drive.py index 523354d05..248dc8b10 100644 --- a/configs/_base_/datasets/drive.py +++ b/configs/_base_/datasets/drive.py @@ -53,5 +53,5 @@ val_dataloader = dict( pipeline=test_pipeline)) test_dataloader = val_dataloader -val_evaluator = dict(type='IoUMetric', iou_metrics=['mIoU']) +val_evaluator = dict(type='IoUMetric', iou_metrics=['mDice']) test_evaluator = val_evaluator diff --git a/configs/_base_/datasets/hrf.py b/configs/_base_/datasets/hrf.py index 3ca2263ab..11b66e7d5 100644 --- a/configs/_base_/datasets/hrf.py +++ b/configs/_base_/datasets/hrf.py @@ -53,5 +53,5 @@ val_dataloader = dict( pipeline=test_pipeline)) test_dataloader = val_dataloader -val_evaluator = dict(type='IoUMetric', iou_metrics=['mIoU']) +val_evaluator = dict(type='IoUMetric', iou_metrics=['mDice']) test_evaluator = val_evaluator diff --git a/configs/_base_/datasets/stare.py b/configs/_base_/datasets/stare.py index a904fa8fd..7fccd71a5 100644 --- a/configs/_base_/datasets/stare.py +++ b/configs/_base_/datasets/stare.py @@ -53,5 +53,5 @@ val_dataloader = dict( pipeline=test_pipeline)) test_dataloader = val_dataloader -val_evaluator = dict(type='IoUMetric', iou_metrics=['mIoU']) +val_evaluator = dict(type='IoUMetric', iou_metrics=['mDice']) test_evaluator = val_evaluator diff --git a/mmseg/datasets/chase_db1.py b/mmseg/datasets/chase_db1.py index 7c697e295..49ef84d53 100644 --- a/mmseg/datasets/chase_db1.py +++ b/mmseg/datasets/chase_db1.py @@ -23,4 +23,4 @@ class ChaseDB1Dataset(BaseSegDataset): seg_map_suffix='_1stHO.png', reduce_zero_label=False, **kwargs) - assert self.file_client.exists(self.data_prefix['img_list']) + assert self.file_client.exists(self.data_prefix['img_path'])