sasaffcxzc 2022-07-04 21:36 采纳率: 63.3%
浏览 283
已结题

请教opencv 运行 onnx模型报错

我用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'


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

  • 写回答

4条回答 默认 最新

  • 爱晚乏客游 2022-07-05 14:14
    关注

    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()  
    
    
    本回答被题主选为最佳回答 , 对您是否有帮助呢?
    评论 编辑记录
查看更多回答(3条)

报告相同问题?

问题事件

  • 系统已结题 7月24日
  • 已采纳回答 7月16日
  • 创建了问题 7月4日

悬赏问题

  • ¥15 请问如何查看手机root记录?
  • ¥15 商城小程序订单号重复
  • ¥15 学校优化算法sbo和蚁群算法怎么结合
  • ¥21 matlab怎么求时域信号的二阶导数
  • ¥15 判断两个表是否完全相同
  • ¥15 java map类型数据格式,如何快速通过前缀匹配元素
  • ¥15 stc12c5a60s2、QMC5883L、LCD1602组合测量磁场所需程序
  • ¥20 Win11测试yolov4,“找不到nvcuda.dll”怎么办?
  • ¥15 simulink绘制bode图
  • ¥15 php_network_getaddresses: getaddrinfo failed: Name or service not known