Как я могу удалить метку класса из обнаруженного объекта с помощью TensorFlow - PullRequest
Новая
       13

Как я могу удалить метку класса из обнаруженного объекта с помощью TensorFlow

2 голосов
/ 27 февраля 2020

Я пытаюсь удалить метку, появляющуюся над окном обнаруженного объекта (в моем случае: Human_hand: 96%). Есть ли у вас какие-либо идеи? Вот как это выглядит:

enter image description here

Я использую следующий код для отображения обнаруженных полей внутри изображения, как я могу отредактировать его, чтобы в состоянии скрыть ярлык? 

while(True):

    # Acquire frame and expand frame dimensions to have shape: [1, None, None, 3]
    # i.e. a single-column array, where each item in the column has the pixel RGB value
    ret, frame = video.read()
    frame_expanded = np.expand_dims(frame, axis=0)

    # Perform the actual detection by running the model with the image as input
    (boxes, scores, classes, num) = sess.run(
        [detection_boxes, detection_scores, detection_classes, num_detections],
        feed_dict={image_tensor: frame_expanded})

    # Draw the results of the detection (aka 'visulaize the results')
    vis_util.visualize_boxes_and_labels_on_image_array(
        frame,
        np.squeeze(boxes),
        np.squeeze(classes).astype(np.int32),
        np.squeeze(scores),
        category_index,
        use_normalized_coordinates=True,
        line_thickness=8,
        min_score_thresh=0.60)

    # All the results have been drawn on the frame, so it's time to display it.
    cv2.imshow('Object detector', frame)

    # Press 'q' to quit
    if cv2.waitKey(1) == ord('q'):
        break

# Clean up
video.release()
cv2.destroyAllWindows()

1 Ответ

3 голосов
/ 27 февраля 2020

Вы можете передать skip_labels=True в vis_util.visualize_boxes_and_labels_on_image_array

...