实例分割Yolact边缘端部署 (三) 从onnx到caffemodel
从ONNX到Caffemodel
- 从目前大多基于pytorch的网络来看,基本都自带onnx导出脚本,这一过程并不复杂,需要考虑一下OP是否是caffe支持的,不支持就需要进行替换,或者基于caffe进行增加对应OP
1. 安装caffe
- 这一步网上教程比较多,按照指导安装即可,这里省略,后面如有需要再进行增加
2. 代码分析
-
简要分析
ONNX2Caffe/convertCaffe.py
中的内容 -
输入onnx路径和caffe输出路径,
getGraph
获取onnx中的网络结构和权重,convertToCaffe()
进行模型转换
onnx_path = "/home/.../ONNX2Caffe/onnx/yolact_sim.onnx"
prototxt_path = "weights/yolcat.prototxt"
caffemodel_path = "weights/yolcat.caff