opencv调用yolov5的onnx模型的常见bug_getlayershapesrecursively

作者：Gausst松鼠会 | 2024-04-24 05:27:08

踩

getlayershapesrecursively

项目场景：在vs/QT中利用opencv调用onnx模型

问题描述

将pt文件转为onnx文件后，利用opencv中的dnn模块去调用出现的问题


[ERROR:0@12.939] global /home/book/lwj/opencv/opencv_ubuntu-4.5.5/modules/dnn/src/dnn.cpp (3875) getLayerShapesRecursively OPENCV/DNN: []:(_input): getMemoryShapes() throws exception. inputs=1 outputs=0/0 blobs=0
[ERROR:0@12.941] global /home/book/lwj/opencv/opencv_ubuntu-4.5.5/modules/dnn/src/dnn.cpp (3878) getLayerShapesRecursively     input[0] = [ 1 3 640 640 ]
[ERROR:0@12.941] global /home/book/lwj/opencv/opencv_ubuntu-4.5.5/modules/dnn/src/dnn.cpp (3888) getLayerShapesRecursively Exception message: OpenCV(4.5.5) /home/book/lwj/opencv/opencv_ubuntu-4.5.5/modules/dnn/src/dnn.cpp:810: error: (-215:Assertion failed) inputs.size() == requiredOutputs in function 'getMemoryShapes'

原因分析：

1：torch版本的问题

楼主之前在使用torch1.18版本时，利用之前的代码转换成功后，opencv能直接调用。但是更新版本后，再重复之前的操作，就会发现出bug了。具体bug如下

[ERROR:0@12.939] global /home/book/lwj/opencv/opencv_ubuntu-4.5.5/modules/dnn/src/dnn.cpp (3875) getLayerShapesRecursively OPENCV/DNN: []:(_input): getMemoryShapes() throws exception. inputs=1 outputs=0/0 blobs=0
[ERROR:0@12.941] global /home/book/lwj/opencv/opencv_ubuntu-4.5.5/modules/dnn/src/dnn.cpp (3878) getLayerShapesRecursively input[0] = [ 1 3 640 640 ]
[ERROR:0@12.941] global /home/book/lwj/opencv/opencv_ubuntu-4.5.5/modules/dnn/src/dnn.cpp (3888) getLayerShapesRecursively Exception message: OpenCV(4.5.5) /home/book/lwj/opencv/opencv_ubuntu-4.5.5/modules/dnn/src/dnn.cpp:810: error: (-215:Assertion failed) inputs.size() == requiredOutputs in function 'getMemoryShapes'

解决方案：

在官网中下载最新的export项目文件，亲测可以成功。

另外由于opencv版本不同，导出onnx文件版本号要与opencv对应。楼主的opencv版本是4.5.5，对应的opset版本为12.

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/Gausst松鼠会/article/detail/477801