从Tensorflow Object Detection API返回Json或其他漂亮的东西

Question

我正在使用tensorflow对象检测API，我想知道在所有python脚本中是否有一个会返回图像中存在哪些标签。喜欢返回Json或类似的东西吗？

Answer 1

没有直接生成JSON的python脚本，但是有一个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

Answer 2

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

这就是您要寻找的。最后一个代码段。