未验证 提交 c1b42028 编写于 作者: W wangguanzhong 提交者: GitHub

fix export model (#507)

上级 922bc008
...@@ -45,9 +45,9 @@ def parse_reader(reader_cfg, metric, arch): ...@@ -45,9 +45,9 @@ def parse_reader(reader_cfg, metric, arch):
if metric == 'COCO': if metric == 'COCO':
from ppdet.utils.coco_eval import get_category_info from ppdet.utils.coco_eval import get_category_info
if metric == "VOC": elif metric == "VOC":
from ppdet.utils.voc_eval import get_category_info 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 from ppdet.utils.widerface_eval_utils import get_category_info
else: else:
raise ValueError( raise ValueError(
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册