Output score , class and id Extraction using TensorFlow object detection

前端 未结 3 1178
执念已碎
执念已碎 2021-01-25 21:33

How can I extract the output scores for objects , object class ,object id detected in images , generated by the Tensorflow Model for Object Detection ?

I want to store

3条回答
  •  半阙折子戏
    2021-01-25 22:17

    It gives you the class with the highest score because output tensors are sorted from highest score to lowest and you are asking for the highest score by indexing to the first element [0].

    Look at object_detection/inference/detection_inference for inspiration.

    As for class names, you can use the label map to create a category index dictionary to translate class ids to names.

提交回复
热议问题