Search code examples
tensorflowobject-detectionobject-detection-apitensorflow-hub

Print detected classes after using Tensorflow Hub for an Object Detection


I would like to use the Tensorflow Hub Example Code for a little Project. I don't know how to write a function to print out all the selected classes and scores after the output of the image with boxes. Can someone help me with that?


Solution

  • Okay I found the solution.

    You have to edit the "draw_boxes" function and add a "object_list.append(display_str)" function in the for loop & then print the object_list after the loop