未验证 提交 b7c50f5b 编写于 作者: G Guanghua Yu 提交者: GitHub

fix oid model export model (#3037)

上级 d7ff713e
......@@ -67,6 +67,8 @@ def parse_reader(reader_cfg, metric, arch):
from ppdet.utils.voc_eval import get_category_info
elif metric == "WIDERFACE":
from ppdet.utils.widerface_eval_utils import get_category_info
elif cfg.metric == 'OID':
from ppdet.utils.oid_eval import get_category_info
else:
raise ValueError(
"metric only supports COCO, VOC, WIDERFACE, but received {}".format(
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册