From c1b42028c88509c7eab9d7a3f18ee4377aceb5b6 Mon Sep 17 00:00:00 2001 From: wangguanzhong Date: Wed, 15 Apr 2020 16:55:27 +0800 Subject: [PATCH] fix export model (#507) --- tools/export_model.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tools/export_model.py b/tools/export_model.py index cba13896f..f92031f6b 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( -- GitLab