[Fix] 2.0 Fix mDice metric for medical datasets (#1864)
parent
e2f26e3fde
commit
e4b9d72a11
|
@ -55,5 +55,5 @@ val_dataloader = dict(
|
||||||
pipeline=test_pipeline))
|
pipeline=test_pipeline))
|
||||||
test_dataloader = val_dataloader
|
test_dataloader = val_dataloader
|
||||||
|
|
||||||
val_evaluator = dict(type='IoUMetric', iou_metrics=['mIoU'])
|
val_evaluator = dict(type='IoUMetric', iou_metrics=['mDice'])
|
||||||
test_evaluator = val_evaluator
|
test_evaluator = val_evaluator
|
||||||
|
|
|
@ -53,5 +53,5 @@ val_dataloader = dict(
|
||||||
pipeline=test_pipeline))
|
pipeline=test_pipeline))
|
||||||
test_dataloader = val_dataloader
|
test_dataloader = val_dataloader
|
||||||
|
|
||||||
val_evaluator = dict(type='IoUMetric', iou_metrics=['mIoU'])
|
val_evaluator = dict(type='IoUMetric', iou_metrics=['mDice'])
|
||||||
test_evaluator = val_evaluator
|
test_evaluator = val_evaluator
|
||||||
|
|
|
@ -53,5 +53,5 @@ val_dataloader = dict(
|
||||||
pipeline=test_pipeline))
|
pipeline=test_pipeline))
|
||||||
test_dataloader = val_dataloader
|
test_dataloader = val_dataloader
|
||||||
|
|
||||||
val_evaluator = dict(type='IoUMetric', iou_metrics=['mIoU'])
|
val_evaluator = dict(type='IoUMetric', iou_metrics=['mDice'])
|
||||||
test_evaluator = val_evaluator
|
test_evaluator = val_evaluator
|
||||||
|
|
|
@ -53,5 +53,5 @@ val_dataloader = dict(
|
||||||
pipeline=test_pipeline))
|
pipeline=test_pipeline))
|
||||||
test_dataloader = val_dataloader
|
test_dataloader = val_dataloader
|
||||||
|
|
||||||
val_evaluator = dict(type='IoUMetric', iou_metrics=['mIoU'])
|
val_evaluator = dict(type='IoUMetric', iou_metrics=['mDice'])
|
||||||
test_evaluator = val_evaluator
|
test_evaluator = val_evaluator
|
||||||
|
|
|
@ -23,4 +23,4 @@ class ChaseDB1Dataset(BaseSegDataset):
|
||||||
seg_map_suffix='_1stHO.png',
|
seg_map_suffix='_1stHO.png',
|
||||||
reduce_zero_label=False,
|
reduce_zero_label=False,
|
||||||
**kwargs)
|
**kwargs)
|
||||||
assert self.file_client.exists(self.data_prefix['img_list'])
|
assert self.file_client.exists(self.data_prefix['img_path'])
|
||||||
|
|
Loading…
Reference in New Issue