mirrors/mmdeploy
1
0
Fork 0
mirror of https://github.com/open-mmlab/mmdeploy.git synced 2025-01-14 08:09:43 +08:00
Code Issues Projects Releases Wiki Activity

concat datasets pytorch metric (#1447)

Browse Source
This commit is contained in:
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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
tests/regression/mmocr.yml
View File

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

2
tools/regression_test.py
View File

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