现在,我正在编写一个程序来检测和计数图像中的对象。我的代码基于this code。
检测图像中的对象没有问题,但是找不到找到它们的方法。
有人说,添加
print([category_index.get(i) for i in classes[0]])
print(scores)
在object_detection_image.py
但是什么也没发生。
def visualize_boxes_and_labels_on_image_array(
image,
boxes,
classes,
scores,
category_index,
instance_masks=None,
instance_boundaries=None,
keypoints=None,
use_normalized_coordinates=False,
max_boxes_to_draw=20,
min_score_thresh=.5,
agnostic_mode=False,
line_thickness=4,
groundtruth_box_visualization_color='black',
skip_scores=False,
skip_labels=False,
class_name(print(class_name)):
答案 0 :(得分:0)
您可以通过box的.shape属性找到它,然后找到该形状的长度
print(len(boxes.shape))
获取箱数