请教opencv 运行 onnx模型报错

Question

我用yolov5.6.0的5s模型转换成onnx模型。
然后根据网上 opencv教程部署，可是到了最后一步 net.forward() 就报错了

import cv2
net = cv2.dnn.readNetFromONNX("yolov5s.onnx") 
image = cv2.imread("11.bmp")  # 读取图片
print(image)
blob = cv2.dnn.blobFromImage(image)  
out = net.forward()  

报错代码：

cv2.error: OpenCV(4.5.5) D:\a\opencv-python\opencv-python\opencv\modules\dnn\src\dnn.cpp:3457: error: (-215:Assertion failed) !layers[0].outputBlobs.empty() in function 'cv::dnn::dnn4_v20211220::Net::Impl::allocateLayers'

请教一下什么原因引起的，翻了很多关于报错这方面，有的说图片，有的说版本，有的说路径，都试过了也一样

Answer 1

yolov5 6.0以上的版本是可以直接读取onnx的，你这里少了一句setInput而已，另外，blobFromImage需要指定onnx导出的网络宽度大小，不一致的话forward会报错。后处理部分可以看下yolov5的detect.py的源码

import cv2
net = cv2.dnn.readNetFromONNX("yolov5s.onnx") 
image = cv2.imread("11.bmp")  # 读取图片
print(image)
blob = cv2.dnn.blobFromImage(image,1/255.0,(640,640),(0,0,0),True,False)  
net.setInput(blob)
out = net.forward()