Ограничительные рамки не отображаются для модели FasterRcnn - PullRequest
0 голосов
/ 01 мая 2020

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

Исходный код

def draw_boxes(result,imgToDisplay,w,h):

    #Draw bounding boxes onto the frame.
    for box in result[0][0]: # Output shape is 1x1x100x7
            conf = box[2]  #box[2] = result[0][0][2] is a probability value
            print("probability value : ",conf)
            print("box[0] : ",box[0]) #Object found or not
            print("box[1] : ",box[1]) #Object class
            print("xmin : ", box[3])
            print("ymin : ", box[4])
            print("xmax : ", box[5])
            print("ymax : ", box[6])

            #In below code, w and h are height and width of image respectively
            if conf >= 0.1:
                xmin = int(box[3] * w)  
                ymin = int(box[4] * h) 
                xmax = int(box[5] * w) 
                ymax = int(box[6] * h) 
                cv2.rectangle(imgToDisplay, (xmin, ymin), (xmax, ymax), (0, 0, 255), 1)

             cv2.imshow('Object_detection',imgToDisplay)
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...