从 tensorflow 对象检测 API 返回 Json 或漂亮的东西

Question

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

Answer 1

没有直接生成 JSON 的 python 脚本，但是有一个 Jupyter notebook 可以演示如何进行推理：

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/

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