Fix regression test for mmocr (#2099)

* fix text recog regression test

* update mmocr.yml and change | to + in regression_test.py

* recover
pull/2105/head
AllentDan 2023-05-22 16:09:11 +08:00 committed by GitHub
parent 11af78caae
commit 8670d838cb
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 11 additions and 4 deletions

View File

@ -10,10 +10,18 @@ globals:
hmean-iou: # named after metafile.Results.Metrics
metric_key: icdar/hmean # eval key name
tolerance: 0.15 # metric ±n%
word_acc:
IIIT5K word_acc:
metric_key: IIIT5K/recog/word_acc_ignore_case_symbol
tolerance: 0.05 # metric ±n%
dataset: IIIT5K
SVT word_acc:
metric_key: SVT/recog/word_acc_ignore_case_symbol
tolerance: 0.05 # metric ±n%
dataset: SVT
SVTP word_acc:
metric_key: SVTP/recog/word_acc_ignore_case_symbol
tolerance: 0.05 # metric ±n%
dataset: SVTP
convert_image_det: &convert_image_det
input_img: *img_densetext_det
test_img: *img_demo_text_det
@ -71,7 +79,6 @@ tensorrt:
pipeline_trt_detection_dynamic_fp32: &pipeline_trt_detection_dynamic_fp32
convert_image: *convert_image_det
backend_test: *default_backend_test
sdk_config: *sdk_detection_dynamic
deploy_config: configs/mmocr/text-detection/text-detection_tensorrt_dynamic-320x320-2240x2240.py
pipeline_trt_detection_mrcnn_dynamic_fp32: &pipeline_trt_detection_mrcnn_dynamic_fp32

View File

@ -314,8 +314,8 @@ def get_pytorch_result(model_name: str, meta_info: dict, checkpoint_path: Path,
using_task.add(task_name)
using_dataset.add(dataset)
dataset_type = '|'.join(list(using_dataset))
task_type = '|'.join(list(using_task))
dataset_type = '+'.join(list(using_dataset))
task_type = '+'.join(list(using_task))
metric_list = []
for metric, metric_info in test_yaml_metric_info.items():
value = '-'