yolo模型转换：pytorch -＞ onnx -＞ caffe

第一步：pytorch转onnx（pytorch版，yolov3-9.0开始提供脚本export.py）

（1）设置onnx算子版本（按需）

  修改代码：
torch.onnx.export(model, img, f, verbose=False, opset_version=12, input_names=['images'],为：
torch.onnx.export(model, img, f, verbose=True, opset_version=9, input_names=['images'], do_constant_folding=True,

（2）导出onnx

python3 ./models/export.py --weights ./weights/yolov3.pt

（3）onnx模型简化，项目地址： https://github.com/daquexian/onnx-simplifier，否则下一步报错TypeError: ONNX node of type Shape is not supported.

pip install onnx-simplifierpython3 -m onnxsim ./weights/yolov3.onnx ./weights/yolov3-si.onnx

第二步：onnx转caffe

git clone https://github.com/xxradon/ONNXToCaffe.gitcd ONNXToCaffepython3 convertCaffe.py ./model/yolov3-si.onnx ./model/yolov3-si.prototxt ./model/yolov3-si.caffemodel

报错一：

Traceback (most recent call last):File "convertCaffe.py", line 122, in <module>convertToCaffe(graph, opset_version, prototxt_path, caffemodel_path)File "convertCaffe.py", line 67, in convertToCaffelayer = converter_fn(node,graph,err)File "/home/ubuntu/ONNXToCaffe-master/onnx2caffe/_operators.py", line 257, in _convert_Reshapereturn err.unsupported_op_configuration(node, "Reshape dimention number shall be 2 or 4")File "/home/ubuntu/ONNXToCaffe-master/onnx2caffe/_error_utils.py", line 44, in unsupported_op_configurationraise TypeError(
TypeError: Error while converting op of type: Reshape. Error message: Reshape dimention number shall be 2 or 4

修改_operators.py（yolo的数据shape是5维数组）

vim ./onnx2caffe/_operators.py修改代码：
elif len(shape) == 4:为
elif len(shape) == 4 or len(shape) == 5:

报错二：AttributeError: permute_param 或者 AttributeError: Upsample_param

Traceback (most recent call last):File "convertCaffe.py", line 122, in <module>convertToCaffe(graph, opset_version, prototxt_path, caffemodel_path)File "convertCaffe.py", line 79, in convertToCaffelayers[id] = layer._to_proto()File "/home/ubuntu/ONNXToCaffe-master/MyCaffe.py", line 100, in _to_protoassign_proto(layer, k, v)File "/home/ubuntu/ONNXToCaffe-master/MyCaffe.py", line 29, in assign_protois_repeated_field = hasattr(getattr(proto, name), 'extend')
AttributeError: permute_param

添加Upsample层、permute层，重新编译caffe，项目地址：https://github.com/jnulzl/caffe_plus

git clone https://github.com/jnulzl/caffe_plus.gitcp caffe_plus/include/caffe/layers/upsample_layer.hpp caffe/include/caffe/layers/
cp caffe_plus/src/caffe/layers/upsample_layer.cpp caffe/src/caffe/layers/
cp caffe_plus/src/caffe/layers/upsample_layer.cu caffe/src/caffe/layers/
cp caffe_plus/include/caffe/layers/permute_layer.hpp caffe/include/caffe/layers/
cp caffe_plus/src/caffe/layers/permute_layer.cpp caffe/src/caffe/layers/
cp caffe_plus/src/caffe/layers/permute_layer.cu caffe/src/caffe/layers/# 修改caffe.proto文件vim caffe/src/caffe/proto/caffe.proto在optional WindowDataParameter window_data_param = 129;（约第423行）后增加代码：optional PermuteParameter permute_param = 150;
optional UpsampleParameter upsample_param = 151;在末尾增加代码：
message PermuteParameter {// The new orders of the axes of data. Notice it should be with// in the same range as the input data, and it starts from 0.// Do not provide repeated order.repeated uint32 order = 1;
}message UpsampleParameter {        optional int32 height = 1 [default = 32];optional int32 width = 2 [default = 32];optional int32 height_scale = 3 [default = 2];optional int32 width_scale = 4 [default = 2];enum UpsampleOp {NEAREST = 0;BILINEAR = 1;}optional UpsampleOp mode = 5 [default = BILINEAR];
}# 重新编译cd caffe
make all -j8
make pycaffe -j8

相关文章：一文带你熟悉Pytorch-＞Caffe-＞om模型转换流程，海思开发：yolo v5s ：pytorch-＞onnx-＞caffe-＞nnie

相关项目：https://github.com/Wulingtian/yolov5_onnx2caffe

yolo模型转换：pytorch -＞ onnx -＞ caffe相关推荐

darknet cpp weights模型转换成ONNX模型
整理不易,如果觉得有用,记得点赞收藏和分享哦 1. 下载转换需要的代码文件在下面地址下载代码文件 https://gitee.com/liangjiaxi2019/pytorch-YOLOv4 2. ...
pytorch模型(.pt)转onnx模型(.onnx)的方法详解(1)
1. pytorch模型转换到onnx模型 2.运行onnx模型 3.比对onnx模型和pytorch模型的输出结果我这里重点是第一点和第二点,第三部分比较容易首先你要安装依赖库:onnx ...
轻松入门模型转换和可视化
点击上方"3D视觉工坊",选择"星标" 干货第一时间送达本文给大家介绍一个模型转换格式ONNX和可视化工具Netron.ONNX是微软设计的一种多平台的通用文 ...
模型转换、压缩、加速工具
20210618 sky_hole: 回成都工作了吗?wang shi yang: 嗯我现在已经在成都上班了sky_hole: 不用付费,我之前发你的视频你好好看看就可以入门了sky_hole: 成 ...
模型转换、模型压缩、模型加速工具汇总
点击上方"计算机视觉工坊",选择"星标" 干货第一时间送达编辑丨机器学习AI算法工程一.场景需求解读在现实场景中,我们经常会遇到这样一个问题,即某篇论 ...
将训练好的pytorch模型的pth文件转换成onnx模型（亲测成功）
将训练好的pytorch模型的pth文件转换成onnx模型(亲测成功) 模型转换声明:本文原创,未经许可严禁转载,原文地址https://blog.csdn.net/hutao1030813002/ ...
linux caffe生成的模型,深度学习之pytorch转caffe转ncnn模型转换（三）
搭建caffe平台: 先在Linux系统下搭建caffe环境,安装依赖包: sudo apt-get install libprotobuf-dev libleveldb-dev libsnappy- ...
【地平线开发板模型转换】将pytorch生成的onnx模型转换成.bin模型
文章目录 1 获取onnx模型 2 启动docker容器 3 onnx模型检查 3.1 为什么要检查? 3.2 如何操作 4 图像数据预处理 4.1 一些问题的思考 4.2 图片挑选与放置 4.2 使 ...
pytorch模型转ONNX转TensorRT，模型转换和推理部署
一.pth模型转ONNX import os import sys import torch import numpy as npfrom feat.model import ResNet # 导入自 ...

yolo模型转换：pytorch -＞ onnx -＞ caffe

yolo模型转换：pytorch -＞ onnx -＞ caffe相关推荐

最新文章

热门文章