Ошибка IndexError: индекс 14708 выходит за пределы для оси 0 с размером 295

#python #opencv #yolo

Вопрос:

Я пытаюсь создать программное обеспечение для обнаружения объектов с помощью yolo, и появляется эта ошибка, и я так потерян, пожалуйста, кто-нибудь может мне помочь!! (код не завершен, и извините, если в этом сообщении есть какие-либо ошибки, потому что я новичок в Stackoverflow).учебное пособие взято отсюда

 Traceback (most recent call last):
  File "d:/opencv/objdetect_yolo.py", line 66, in <module>
    findobj(output,img)
  File "d:/opencv/objdetect_yolo.py", line 33, in findobj
    cofidence = scores[classId]
IndexError: index 14708 is out of bounds for axis 0 with size 295

Ошибка IndexError: индекс 14708 выходит за пределы для оси 0 с размером 295

 import numpy as np 
import cv2

cap = cv2.VideoCapture(0)
whT = 320

classespath = 'coco.names.txt'
classes = []

with open(classespath,'rt')as f:
    classes = f.read().rstrip('n').split('n')
#print (classes)
#print(len(classes))

modelConfiguration = 'yolov3.cfg'
modelWeights = 'yolov3.weights'

net = cv2.dnn.readNetFromDarknet(modelConfiguration, modelWeights)
net.setPreferableBackend(cv2.dnn.DNN_BACKEND_OPENCV)
net.setPreferableTarget(cv2.dnn.DNN_TARGET_CPU)

def findobj(outputs,img):
    hT, wT , cT = img.shape
    bbox = []
    classIds = []
    confs = []


    for output in outputs:
        for det in outputs:
            scores = det[5:]
            classId = np.argmax(scores)
            cofidence = scores[classId]
            if float(0.5) < cofidence:

            
                w,h = int(det[2]*wT),int(det[3]*hT)
                x,y = int((det[0]*wT) - w/2), int((det[1]*hT) - h/2)
                bbox.append([x,y,w,h])
                classIds.append(classId)
                confs.append(float(cofidence))
              




     
while True:
    succes, img = cap.read()

    blob = cv2.dnn.blobFromImage(img,1/255,(whT,whT),[0,0,0],1,crop=False)
    net.setInput(blob)

    layerNames = net.getLayerNames()
    #print(layerNames)
    outputNames = [layerNames[i[0]-1]for i in net.getUnconnectedOutLayers() ]
    #print(outputNames)
    #print(net.getUnconnectedOutLayers())
    output = net.forward(outputNames)


    findobj(output,img)


    cv2.imshow("objdetect",img)
    if cv2.waitKey(1) amp; 0xFF == ord('q'):
        break

1. пожалуйста, кто-нибудь может мне помочь с этим

2. Добро пожаловать в SO. Пожалуйста, укажите полную ошибку обратной трассировки.

3. Добро пожаловать в SO. Пожалуйста, укажите полную ошибку обратной трассировки — как???

4. Отредактируйте свое сообщение повторно, скопируйте и вставьте полную ошибку обратной трассировки в сообщение.

5. Обратная трассировка (последний последний вызов): файл «d:/opencv/objdetect_yolo.py «, строка 66, в файле <module> findobj(output,img) «d:/opencv/objdetect_yolo.py «, строка 33, в findobj cofidence = scores[classId] Ошибка индекса: индекс 14708 выходит за пределы оси 0 с размером 295

Ответ №1:

Похоже, вы столкнулись с проблемой, потому np.argmax что вместо индекса вам будет указан необработанный номер элемента max. Итак, если у вас есть матрица 3×3, функция argmax будет обрабатывать матрицу как строку 9×1 вместо квадрата 3×3.

 # The matrix:
[[1, 2, 3],
 [4, 5, 6],
 [7, 8, 9]]

#will be treated as:
[1, 2, 3, 4, 5, 6, 7, 8, 9]

В документации предлагается следующее решение:

 classId = np.unravel_index(np.argmax(scores, axis=None), scores.shape)