TypeError cv2.rect - PullRequest
Новая
       32

TypeError cv2.rect

1 голос
/ 19 апреля 2020

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

1) ssd_inception_v2_coco_2018_01_28 модель обнаружения объекта. Успешно преобразовано в IR.

2) Снимите входное изображение и предварительно обработайте его в соответствии с информацией о входном слое модели.

3) Успешно извлекли выходные данные модели.

4) Добавлены ограничительные рамки в выходное изображение.

Но я получаю следующую ошибку, пожалуйста, помогите!



Исходный код 

#Preprocessing image

img = r'D:\SriSailam.jpg'
img = cv2.imread(img)
img = cv2.resize(img,(300,300))
img = img.transpose((2,0,1))
img = img.reshape(1,3,300,300)



exec_net = engine.load_network(net,'cpu')


#Asynchronous request

infer_request_handle = exec_net.start_async(request_id=0, inputs={'image_tensor': img}) 
infer_status = infer_request_handle.wait()
output = next(iter(net.outputs))


n,c,h,w = img.shape

result = infer_request_handle.outputs[output]  #output layer shape


'''
Draw bounding boxes onto the image.
'''
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])
        print("box[1] : ",box[1])
        print("xmin : ", box[3])
        print("ymin : ", box[4])
        print("xmax : ", box[5])
        print("ymax : ", box[6])

        if conf >= 0:
            xmin = int(box[3] * w)  #box[3] - xmin
            ymin = int(box[4] * h) #box[4] - ymin
            xmax = int(box[5] * w)  #box[5] - xmax
            ymax = int(box[6] * h) #box[6] - ymax
            rect = cv2.rectangle(img, ((xmin), (ymin)), ((xmax), (ymax)), (0, 0, 255), 1)
cv2.imshow(rect,'face_detection')
cv2.waitKey(0)



Выход 

C:\Users\adity\Desktop\openvino>py check_infer.py                                                                                                                        
probability value :  0.6344592                                                                                                                                           
box[0] :  0.0                                                                                                                                                            
box[1] :  1.0                                                                                                                                                            
xmin :  0.1831626                                                                                                                                                        
ymin :  0.10697469                                                                                                                                                       
xmax :  0.86960804                                                                                                                                                       
ymax :  1.0                                                                                                                                                              
Traceback (most recent call last):                                                                                                                                       
  File "check_infer.py", line 68, in <module>                                                                                                                            
    rect = cv2.rectangle(img, ((xmin), (ymin)), ((xmax), (ymax)), (0, 0, 255), 1)                                                                                        
TypeError: an integer is required (got type tuple)

1 Ответ

1 голос
/ 19 апреля 2020

Воспроизведена ваша ошибка. Сообщение об ошибке довольно запутанно в этой версии OpenCV (4.2.0).

Если вы сделаете то же самое в OpenCV 4.3.0, вы получите более понятную ошибку, такую ​​как 

  File "<stdin>", line 1, in <module>
TypeError: Expected Ptr<cv::UMat> for argument 'img'

, которая в основном говорит, что аргумент img, который вы передали в cv2.rectangle, имеет неверную тип. Проблема в том, что после того, как вы позвоните 

img = img.transpose((2,0,1))
img = img.reshape(1,3,300,300)

img, OpenCV не сможет интерпретировать его как изображение. Проблема может быть решена, например, следующим образом:
1. Скопируйте img перед преобразованием в другую переменную imgForDisplaying
2. Используйте imgForDisplaying, чтобы нарисовать прямоугольник. 

img = r'D:\SriSailam.jpg'
img = cv2.imread(img)
img = cv2.resize(img,(300,300))
imgToDisplay = img
img = img.transpose((2,0,1))
img = img.reshape(1,3,300,300)
...
rect = cv2.rectangle(imgToDisplay, (xmin, ymin), (xmax, ymax), (0, 0, 255), 1)
...