[Fix] 2.0 Fix mDice metric for medical datasets (#1864)
parent
e2f26e3fde
commit
e4b9d72a11
|
@ -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
|
||||
|
|
|
@ -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
|
||||
|
|
|
@ -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
|
||||
|
|
|
@ -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
|
||||
|
|
|
@ -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'])
|
||||
|
|
Loading…
Reference in New Issue