caffe2的发布,其优点有:1支持大规模分布式计算;2.支持移动部署,,,,等原因,而且据说caffe2的运行速度相对于caffe速度会快很多,所以因为一些业务需要尝试将caffe转caffe2,在转的过程中遇到很多坑,现分享一下自己成功转换的步骤,希望能帮到您
首先安装好caffe2,相关安装教程请参考官方文档:https://caffe2.ai/docs/getting-started.html?platform=mac&configuration=compile
要注意如果使用的是虚拟环境或者其他,需要把改caffe2安装在对应的环境下。
安装好之后就可以开始转换之旅啦。
1.在命令行中激活自己python运行的环境,本小白使用的是pycharm,并将caffe2安装在对应的虚拟环境下。
本人虚拟环境为
激活虚拟环境:source tensorflow_pycharm/bin/activate
2.caffe转caffe2我是参考的官方说明:https://caffe2.ai/docs/caffe-migration.html
2.1. 配置需转换的caffe文件,具体为在自己的工程中的caffe文件夹中的models文