concat datasets pytorch metric (#1447)
parent
d77aeaa480
commit
d9d3ded8bc
|
@ -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
|
||||
|
|
|
@ -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']
|
||||
|
|
Loading…
Reference in New Issue