concat datasets pytorch metric (#1447)

pull/1451/head
AllentDan 2022-11-28 17:34:02 +08:00 committed by GitHub
parent d77aeaa480
commit d9d3ded8bc
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 3 additions and 0 deletions

View File

@ -16,6 +16,7 @@ globals:
metric_key: 0_word_acc_ignore_case
tolerance: 1.0 # metric
multi_value: 100
dataset: IIIT5K
convert_image_det: &convert_image_det
input_img: *img_densetext_det
test_img: *img_demo_text_det

View File

@ -314,6 +314,8 @@ def get_pytorch_result(model_name: str, meta_info: dict, checkpoint_path: Path,
for ds in configured_dataset:
if ds in _metrics:
pytorch_metric.update(_metrics[ds])
if ds == dataset:
pytorch_metric.update(_metrics)
else:
pytorch_metric.update(_metrics)
task_name = metafile_metric['Task']