如何将检测到的对象转换为 COCO 数据集 Json

Question

我在https://github.com/tensorflow/models/blob/master/research/object_detection/object_detection_tutorial.ipynb关注对象检测演示：

# Actual detection.
output_dict = run_inference_for_single_image(image_np, detection_graph) 

但我想将 output_dict（函数 run_inference_for_single_image(image_np, detection_graph) 的输出）转换为 COCO 注释 JSON 类型，以便我可以输入它以在不同的对象检测模型之间进行基准测试。

这是基准模型的代码：https://github.com/cocodataset/cocoapi/blob/master/PythonAPI/pycocoEvalDemo.ipynb

#initialize COCO detections api
resFile='%s/results/%s_%s_fake%s100_results.json'
resFile = resFile%(dataDir, prefix, dataType, annType)
cocoDt=cocoGt.loadRes(resFile)

但是你需要输入一个COCO Json类型。

有人可以告诉我如何从 output_dict 转换为 COCO Json 吗？

Answer 1

这就是你要找的。最后的代码sn -p：

https://lijiancheng0614.github.io/2017/08/22/2017_08_22_TensorFlow-Object-Detection-API/