解决onnx转ncnn时Squeeze not supported yet!

出现这个问题是ncnn不支持squeeze操作导致的，
第一步做法：参考https://github.com/daquexian/onnx-simplifier尝试解决，安装后，命令：
python -m onnxsim ./your_model.onnx ./your_model_sim.onnx
如果上面的做法解决不了你的问题，你需要修改你的网络结构的代码，
以pytorch为例：
如果你的某层网络输出数据维度是（1，1280，1，1）接下来需要变为（1，1280）
你的代码是这样写的
x=x.squeeze(-1).squeeze(-1)
只需要改为
x.view(size(1), -1)
其他的情况以此类推。

写成shell脚本方便使用

#!/bin/shif [ $# -ne 2 ];then
echo "./onnxsim.sh [onnx_model_name] [onnxsim_model_name]"
exit -1
fipython -m onnxsim $1 $2

Squeeze not supported yet!相关推荐

树莓派4B使用NCNN部署Yolov5-lite
目录文章目录前言一.树莓派配置NCNN 1.安装依赖 2.下载NCNN并编译二.Yolov5-lite模型训练 1.源码地址 2.安装所需要的包 3.训练自己的数据集(YOLO格式) 4.模型 ...
pandas读取大csv报错：TypeError: ufunc ‘isnan‘ not supported for the input types, and the inputs could not
使用pandas读取csv: BPIC_2017_df = pd.read_csv('./datasets_csv/BPI_Challenge_2017.csv', index_col=0) 但是由于 ...
RuntimeError: 0D or 1D target tensor expected, multi-target not supported解决
loss_class = nn.NLLLoss() loss_class_test = loss_class(class_prd,labels) 报错:RuntimeError: 0D or 1D t ...
RuntimeError: 1only batches of spatial targets supported (non-empty 3D tensors) but got targets of s
目录在运行UNet3+进行多分类训练的时候报错 RuntimeError: 1only batches of spatial targets supported (non-empty 3D tens ...
解决pip安装报错：is not a supported wheel on this platform
原因1 你下载安装的包不是当前平台所支持的原因2 你下载的包,不符合你所在的平台的安装whl的名称规范,所以出错. 比如当前我要安装的包是:PyMuPDF-1.16.16-cp36-cp36m-ma ...
is not a supported wheel on this platform pip安装报错
原因1 你下载安装的包不是当前平台所支持的原因2 你下载的包,不符合你所在的平台的安装whl的名称规范,所以出错. 比如当前我要安装的包是:PyMuPDF-1.16.16-cp36-cp36m-ma ...
pytorch学习中 torch.squeeze() 和torch.unsqueeze()的用法
squeeze的用法主要就是对数据的维度进行压缩或者解压. 先看torch.squeeze() 这个函数主要对数据的维度进行压缩,去掉维数为1的的维度,比如是一行或者一列这种,一个一行三列(1,3)的 ...
python读取xml文件报错ValueError: multi-byte encodings are not supported
1.在使用python对xml文件进行读取时,提示ValueError: multi-byte encodings are not supported 很多贴子上说把xml的编码格式改为,就可以正常执 ...
pytorch之expand，gather，squeeze，sum，contiguous，softmax，max，argmax
目录 gather squeeze expand sum contiguous softmax max argmax gather torch.gather(input,dim,index,out=N ...

Squeeze not supported yet!

解决onnx转ncnn时Squeeze not supported yet!

Squeeze not supported yet!相关推荐

最新文章

热门文章