From 42bd90d717085b2fc8d5f718def120dc617b32b9 Mon Sep 17 00:00:00 2001 From: Guanghua Yu <742925032@qq.com> Date: Fri, 15 May 2020 10:50:59 +0800 Subject: [PATCH] fix catid2name in coco eval (#681) --- ppdet/utils/coco_eval.py | 1 + 1 file changed, 1 insertion(+) diff --git a/ppdet/utils/coco_eval.py b/ppdet/utils/coco_eval.py index 8fde61de6..b065a276a 100644 --- a/ppdet/utils/coco_eval.py +++ b/ppdet/utils/coco_eval.py @@ -609,6 +609,7 @@ def coco17_category_info(with_background=True): if not with_background: clsid2catid = {k - 1: v for k, v in clsid2catid.items()} + catid2name.pop(0) else: clsid2catid.update({0: 0}) -- GitLab