coc预测格式数据类别与训练json中定义的对应不上
Created by: L-lei
在ools/infer.py预测数据后保存的图片与训练数据anno_path变量对应的json文件中类别对应不上； 我的解决如下： 第1步.修改ppdet/data/source/coco.py中87行catid2clsid为： cats = coco.loadCats(coco.getCatIds()) cats.sort(key=lambda a: a['id']) catid2clsid = { cat['id']: i + int(self.with_background) for i, cat in enumerate(cats) } 第2步./home/yt/cl/project_all/PaddleDetection/ppdet/utils/coco_eval.py中 423行clsid2catid 变量之前添加排序，cats.sort(key=lambda a:a['id'])