diff --git a/tools/export_model.py b/tools/export_model.py index cba13896fda4a0daacba414e216b6a16cf7011e8..f92031f6b789ecbbc2a14d3d05857bd5e8f7a425 100644 --- a/tools/export_model.py +++ b/tools/export_model.py @@ -45,9 +45,9 @@ def parse_reader(reader_cfg, metric, arch): if metric == 'COCO': from ppdet.utils.coco_eval import get_category_info - if metric == "VOC": + elif metric == "VOC": from ppdet.utils.voc_eval import get_category_info - if metric == "WIDERFACE": + elif metric == "WIDERFACE": from ppdet.utils.widerface_eval_utils import get_category_info else: raise ValueError(