最近很多朋友留言问我如何将检测结果写入json文件并且用于COCO API的评估,之前对于检测结果的格式已经做了简单的说明,这里提供一些简单的函数,直接调用将结果写入即可。
用于COCO API测试的文件格式
HUST小菜鸡:用于COCO API测试的结果的文件格式zhuanlan.zhihu.com使用COCO API进行结果评估
HUST小菜鸡:使用COCO API评估模型在COCO数据集上的结果zhuanlan.zhihu.comCOCO utils给出了一些转换的函数
def det2json(dataset, results):
json_results = []
for idx in range(len(dataset)):
img_id = dataset.img_ids[idx]
result = results[idx]
for label in range(len(result)):
bboxes = result[label]