mmdetection生成测试集的测试结果并保存在.json

mmdetection生成测试集的测试结果并保存在.json

原代码测试验证时没有输出检测结果的文件,只能自己手动改一个了。

本来想自己写一个后来发现原来有,方法如下:
在mmdet/apis/test.py文件中single_gpu_test()函数末尾处,增加以下两句:
    coco=CocoDataset('data/coco/annotations/instances_val2017.json',pipeline=[]) #验证集路径
    json_results=coco._det2json(results=results)
    # write output
    filepath = 'faster_bbox_results.json' #文件需要手动建立
    if os.path.exists(filepath):
        os.remove(filepath)
        json.dump(json_results, open(filepath, 'w'), indent=4)

你可能感兴趣的:(图像处理,深度学习,计算机视觉,目标检测,人工智能)