fix catid2name in coco eval (#681)

42bd90d7 · Guanghua Yu · GitHub · 7bb4fd1b · 42bd90d7
隐藏空白更改
内联并排

Showing with 1 addition and 0 deletion

ppdet/utils/coco_eval.py ppdet/utils/coco_eval.py +1 -0

未找到文件。
--- 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})