关于opencv调用onnx模型的一个错误 Can’t infer a dim denoted by -1 in function ‘cv::dnn::computeShapeByReshapeMas

关于opencv调用onnx模型的一个错误

Can’t infer a dim denoted by -1 in function ‘cv::dnn::computeShapeByReshapeMask’

opencv报错：

Traceback (most recent call last):File "E:/First_graduate/fatigue/Fatigue_mediapipe0622/Fatigue_Mediapipe0622/pth2onnx_test.py", line 40, in <module>outs = net.forward(layerNames)
cv2.error: OpenCV(4.5.5) D:\a\opencv-python\opencv-python\opencv\modules\dnn\src\layers\reshape_layer.cpp:148: error: (-1:Backtrace) Can't infer a dim denoted by -1 in function 'cv::dnn::computeShapeByReshapeMask'

错误大概是网络中的维度问题

错误原因：

输入图像大小与模型的输入大小不一致。
生成onnx时，指定模型输入大小是84x84的

input1 = torch.randn(1, 3, 84, 84)
input_names = ["input"]
output_names = ["output"]
torch.onnx.export(net, input1, "mouthNet2.onnx", verbose=True, input_names=input_names, output_names=output_names)

所以在调用模型推理时输入图像也应当是84x84的。

解决

resize一下即可

img_t = cv.resize(img,(84,84))    #将图片改为模型适用的尺寸

关于opencv调用onnx模型的一个错误 Can’t infer a dim denoted by -1 in function ‘cv::dnn::computeShapeByReshapeMas相关推荐

java调用onnx模型_开源一年多的模型交换格式ONNX，已经一统框架江湖了？
原标题:开源一年多的模型交换格式ONNX,已经一统框架江湖了? 机器之心原创作者:思源近日,微软亚洲研究院和华为举办了 ONNX 合作伙伴研讨会,这是 ONNX 开源社区成立以来首次在中国举办的活 ...
opencv调用yolov3模型来进行图像检测
之前使用了opencv来调用ssd的模型来检测物体,今天学了一下用opencv调用yolov3的模型来检测物体,二者在预测图形的部分,代码流程差不多,反正就是加载模型然后预测输出,但是对于输出结果的处 ...
Opencv中，imag=cv2.cvtColor(imag,cv2.COLOR_BGR2GRAY) 报错：error:!_src.empty() in function ‘cv::cvtColor‘
Opencv,imag=cv2.cvtColor(imag,cv2.COLOR_BGR2GRAY) 报错:error:!_src.empty() in function 'cv::cvtColor' ...
使用opencv dnn 模块调用darknet模型时候出错，不支持relu激活函数
问题: 使用opencv dnn 模块调用darknet模型时候出错,报错信息为不支持relu激活函数以下过程为笔者自己解决该问题的过程,供各位参考学些,因为中间又遇到新的坑,所以各位务必看完再决 ...
记录：：Opencv调用tensorflow2.x-Keras模型
需要用C++调用tensoeflow模型,但我发现现在的tensorflow2.x的版本都是用keras搭建的,不想用动态库,决定直接用Opencv调用模型. 库版本: tensorflow 2.2. ...
opencv调用TensorFlow1.x训练的pb模型
文章目录前言一.搭建虚拟环境二.通过frozen_pb来生成pbtxt文件 1.获取opencv官方的转化文件 2.生成pbtxt文件三.OpenCV调用pb模型前言通过opencv读取p ...
C++：onnxruntime调用FasterRCNN模型
背景: 最近由于项目原因,需要用C++做一些目标检测的任务,就捣鼓一下YOLOv5,发现部署确实很方便,将YOLOv5模型转为onnx模型后,可以用OpenCV的dnn.readNetFromONNX ...
C++利用opencv调用pytorch训练好的分类模型
pytorch保存模型 import torch.onnxd = torch.rand(1, 3, 224, 224,dtype=torch.float,device = 'cuda') m = mo ...
使用OpenCV调用Caffe-SSD训练好的模型
前言 1.OpenCV3之后的dnn模型可以可以调用好多用深度学习框架的训练好的模型. 2.这里我演示是我自己训练的一个围棋棋盘识别模型,使用的神经网络是VGG16,OpenCV3.3,IDE是VS2 ...

关于opencv调用onnx模型的一个错误 Can’t infer a dim denoted by -1 in function ‘cv::dnn::computeShapeByReshapeMas

关于opencv调用onnx模型的一个错误

Can’t infer a dim denoted by -1 in function ‘cv::dnn::computeShapeByReshapeMask’

opencv报错：

错误原因：

解决

关于opencv调用onnx模型的一个错误 Can’t infer a dim denoted by -1 in function ‘cv::dnn::computeShapeByReshapeMas相关推荐

最新文章

热门文章