diff --git a/ppdet/utils/coco_eval.py b/ppdet/utils/coco_eval.py index 8fde61de6e0080929cea3206225e918c542d231b..b065a276a9b06e589db1538cad2a9040b1bf2257 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})