Как преобразовать обнаруженный объект в набор данных 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

Это то, что вы ищете.Последний фрагмент кода:

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