Индекс списка вне диапазона OpenCV Python ⇐ Python

[Anonymous]

В настоящее время я изучаю машинное обучение и работаю над проектом по обнаружению объектов.
Моя проблема в том, что cv2.put_text() не работает должным образом в моем for цикл.
Вот соответствующий код:

Код: Выделить всё

classIds, confs, bbox = net.detect(img, confThreshold=0.5)

if len(classIds) != 0:
for classId, confidence, box in zip(classIds.flatten(), confs.flatten(), bbox):
cv2.rectangle(img, box, (255, 0, 0), thickness=2)
cv2.putText(img, classNames[classId-1].upper(), (box[0]+10, box[1]+30),
cv2.FONT_HERSHEY_COMPLEX, 1, (255, 0, 0), 2)

А это сообщение об ошибке, которое выдает приведенный выше код:

Код: Выделить всё

File "C:\Users\User\projects\opencv\ObjectDetector.py", line 31,
in 
cv2.putText(img, classNames[classId-1].upper(), (box[0]+10, box[1]+30),
~~~~~~~~~~^^^^^^^^^^^
IndexError: list index out of range

Вот мой полный код:
'

Код: Выделить всё

import cv2

# img = cv2.imread('bob.png')

cap = cv2.VideoCapture(0)
cap.set(3, 640)
cap.set(4, 480)

classNames = []
classFile = 'coco.names'
with open(classFile, 'rt') as f:
classNames = f.read().rstrip('\n').split('\n')

configPath = 'ssd_mobilenet_v3_large_coco_2020_01_14.pbtxt'
weightsPath = 'frozen_inference_graph.pb'

net = cv2.dnn.DetectionModel(weightsPath, configPath)
net.setInputSize(320, 320)
net.setInputScale(1.0 / 127.5)
net.setInputMean((127.5, 127.5, 127.5))
net.setInputSwapRB(True)

while True:
succss, img = cap.read()
classIds, confs, bbox = net.detect(img, confThreshold=0.5)
print(classIds,  bbox)

if len(classIds) != 0:
for classId, confidence, box in zip(classIds.flatten(),
confs.flatten(), bbox):
cv2.rectangle(img, box, (255, 0, 0), thickness=2)
cv2.putText(img, classNames[classId-1].upper(),
(box[0]+10, box[1]+30),
cv2.FONT_HERSHEY_COMPLEX, 1, (255, 0, 0), 2)

cv2.imshow('Output', img)
cv2.waitKey(1)

'
Я ожидал, что появится окно с камерой и рамкой вокруг нее, в которой указано имя объекта, но оно просто показало ошибка. Как мне это решить?

Подробнее здесь: https://stackoverflow.com/questions/788 ... ncv-python