Вернуть Json или что-нибудь симпатичное из Tensflow Object Detection API

Question

Я использую API обнаружения объектов tenorflow, и я хотел бы знать, есть ли во всех скриптах python тот, который возвращал бы, какие метки присутствуют на изображении. Например, вернуть Json или что-то в этом роде?

Answer 1

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

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

Answer 2

Нет сценария Python, который напрямую генерирует JSON, но есть блокнот Jupyter, который демонстрирует, как сделать вывод:

https://github.com/tensorflow/models/blob/master/research/object_detection/object_detection_tutorial.ipynb

В качестве альтернативы, если вы хотите использовать ранее существующий сценарий, этот блокнот Jupyter показывает, как использовать сценарий, который генерирует файлы TFRecord: https://github.com/tensorflow/models/blob/master/research/object_detection/g3doc/oid_inference_and_evaluation.md

Который затем может быть обработан с использованием кода в tf_example_decoder.py в качестве ссылки: https://github.com/tensorflow/models/blob/master/research/object_detection/data_decoders/tf_example_decoder.py