Squeeze not supported yet!

解决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/sh

if [ $# -ne 2 ];then
echo "./onnxsim.sh [onnx_model_name] [onnxsim_model_name]"
exit -1
fi

python -m onnxsim $1 $2

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值