[Fix] 2.0 Fix mDice metric for medical datasets (#1864)

pull/1897/head
MengzhangLI 2022-08-04 18:32:52 +08:00 committed by GitHub
parent e2f26e3fde
commit e4b9d72a11
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
5 changed files with 5 additions and 5 deletions

View File

@ -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

View File

@ -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

View File

@ -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

View File

@ -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

View File

@ -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'])