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)

版权声明:本文为CSDN博主「bloomerOAO」的原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接及本声明。
原文链接:https://blog.csdn.net/qq_42744739/article/details/121534817

bloomerOAO

我还没有学会写个人说明!

暂无评论

发表评论

相关推荐

目标检测指标计算

一、指标 True Positive,TP :预测为正样本(Positive),实际为正样本,则True,预测正确。 True Negative,TN &#x

Deep Learning 目标检测

对检测到的结果进行解析  #----------------------------目标检测*解析字典result------------------------------------------- from numpy import arr

YOLOV3预选框验证

对于一个输入图像,比如416*416*3,相应的会输出 13*13*3 26*26*3 52*52*3 10647 个预测框。我们希望这些预测框的信息能够尽量准确的反应出哪些位置存在对象,是哪种对