diff --git a/static/ppdet/utils/export_utils.py b/static/ppdet/utils/export_utils.py index a65fd6e2799edbe0cea361844f550afa8c8515a3..f2e280a8b0c6a474005fc44dbd9a3d5b1fa60976 100644 --- a/static/ppdet/utils/export_utils.py +++ b/static/ppdet/utils/export_utils.py @@ -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(