#python #opencv #conv-neural-network
Вопрос:
я новичок в нейронных сетях. я пытаюсь использовать opencv для обнаружения объектов.я использую предварительно обученную модель yolov3 для обнаружения объектов. этот код отлично работал на моем старом ноутбуке, но недавно я начал использовать новый ноутбук(i5 9-го поколения). и мои старые коды(opencv) не работают. outs=net.forward(output_layers) возвращает значение nan, которое я использую в версии opencv-python 4.5.0. я не в состоянии понять, что пошло не так. я добавляю изображение кода(spyder)и вывод кода, а также список пакетов в моем пакете среды в моей среде #——— мой код #———
import cv2
import numpy as np
net = cv2.dnn.readNet("yolov3.weights", "yolov3.cfg")
classes = []
with open("coco.names", "r") as f:
classes = [line.strip() for line in f.readlines()]
layer_names = net.getLayerNames()
output_layers = [layer_names[i[0] - 1] for i in
net.getUnconnectedOutLayers()]
colors = np.random.uniform(0, 255, size=(len(classes), 3))
img = cv2.imread("spiritualeye.jpg")
img = cv2.resize(img, None, fx=0.4, fy=0.4)
blob=cv2.dnn.blobFromImage(img,0.00392, (416, 416), (0, 0, 0), True,
crop=False)
net.setInput(blob)
outs=net.forward(output_layers)
print(outs)
cv2.imshow('image',img)
cv2.waitKey(0)
cv2.destroyAllWindows()
#-----------------------
Ответ №1:
проблема заключалась в том, что мой файл весов был поврежден, когда я передавал свои данные между ноутбуками. поэтому мне пришлось снова загрузить файлы с весом.