From b7c50f5bd1b8d02661ae3313d5719e5bbaafb6cb Mon Sep 17 00:00:00 2001
From: Guanghua Yu <742925032@qq.com>
Date: Tue, 18 May 2021 09:53:50 +0800
Subject: [PATCH] fix oid model export model (#3037)

---
 static/ppdet/utils/export_utils.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/static/ppdet/utils/export_utils.py b/static/ppdet/utils/export_utils.py
index a65fd6e27..f2e280a8b 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(
-- 
GitLab