TensorFlow Object Detection API print objects found on image to console

前端 未结 6 1776
星月不相逢
星月不相逢 2021-02-01 06:31

I\'m trying to return list of objects that have been found at image with TF Object Detection API.

To do that I\'m using print([category_index.get(i) for

6条回答
  •  猫巷女王i
    2021-02-01 06:55

    From the function signature visualize_boxes_and_labels_on_image_array, you have to set the arguments max_boxes_to_draw, min_score_thresh,

    visualize_boxes_and_labels_on_image_array(image,
                                              boxes,
                                              classes,
                                              scores,
                                              category_index,
                                              instance_masks=None,
                                              keypoints=None,
                                              use_normalized_coordinates=False,
                                              max_boxes_to_draw=20,
                                              min_score_thresh=.5,
                                              agnostic_mode=False,
                                              line_thickness=4)
    

提交回复
热议问题