在实时对象检测(tensorflow)中检测到特定对象类时如何打印任何内容

时间:2019-05-08 19:02:41

标签: python opencv tensorflow real-time

我在树莓派中使用实时对象检测,为此,我的类是“ e浪费”,“有机”,“回收”

我已经使用了这个存储库https://raw.githubusercontent.com/EdjeElectronics/TensorFlow-Object-Detection-on-the-Raspberry-Pi/master/Object_detection_picamera.py

我试图检索类名称,ID,并对在检测到类时如何打印任何内容进行评分,我使用了if条件来检查检测到的类是否是有机的,然后输出“ hello”,但输出为空而没有错误(下面的代码):

绘制检测结果(又称“显示结果”)

    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.40)



    objects = []
    threshold = 0.40 
    for index, value in enumerate(classes[0]):
        object_dict = {}
        if scores[0, index] > threshold:
            object_dict[(category_index.get(value)).get('name').encode('utf8')] = \
                    scores[0, index]
            objects.append(object_dict)

    if objects=='organic':
       print('hello')

当从相机模块中检测到有机分类时,如何打印问候?

0 个答案:

没有答案