前言

Pytorch转Caffe有两种方式，一种是Pytorch直接转Caffe,另一种是Pytorch转Onnx再转Caffe，我这里用的是第二种方式

环境Caffe搭建

不管是第一种方式，还是第二种方式，都需要先搭建Caffe环境，Caffe环境搭建请参考Caffe环境搭建，基于Ubuntu18.04

模型转换

Caffe添加自定义层

这使用caffe时候，转换模型时候遇到了不支持的Layer（Upsample），参考网上大佬们的帖子和代码添加自定义层

• a) 添加头文件include/caffe/layers/upsample_layer.hpp
• b) 添加CPU推理文件src/caffe/layers/upsample_layer.cpp
• c) 添加GPU推理文件src/caffe/layers/upsample_layer.cu
• d) 修改proto定义文件src/caffe/proto/caffe.proto

具体可参考：https://github.com/aidreamwin/caffe.git

编译

sudo make all
sudo make test
sudo make runtest

编译pycaffe

make pycaffe

参考

• PytorchToCaffe
• upsample_layer 1
• upsample_layer 2
• aidreamwin/caffe
• ONNXToCaffe