我正在使用TensorFlow对象检测API,想知道是否有Python脚本能够返回图像中存在哪些标签。例如返回Json格式或类似的东西?
我正在使用TensorFlow对象检测API,想知道是否有Python脚本能够返回图像中存在哪些标签。例如返回Json格式或类似的东西?
目前没有直接生成JSON的Python脚本,但有一个Jupyter笔记本演示了如何进行推理:
如果您想使用现有的脚本,可以使用此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