Как получить Bbox координаты указанного класса c при обнаружении объекта тензорного потока?

Question

Как я могу нарисовать только ящики определенной категории c вместо ящиков всех категорий?

          [boxes, scores, classes, num_detections],
          feed_dict={image_tensor: image_np_expanded})
      # Visualization of the results of a detection.
      # print(scores)
      vis_util.visualize_boxes_and_labels_on_image_array(
          image_np,
          np.squeeze(boxes),
          np.squeeze(classes).astype(np.int32),
          np.squeeze(scores),
          category_index,
          use_normalized_coordinates=True,
          line_thickness=4,
          min_score_thresh=0.9)```