PyTorch模型到Caffe的转换实战指南

avatar
作者
筋斗云
阅读量:0

Pytorch模型转换Caffe模型踩坑指南,代码使用的是Github上的工程,地址:
https://github.com/longcw/pytorch2caffe

操作环境:ubuntu = 14.04          miniconda 3          caffe          pytorch = 0.2.0    torchvision = 0.1.8          python = 2.7   

环境配置:

第一步 : 在miniconda创建一个虚拟环境pytorch2caffe : conda create -n pytorch2caffe python=2.7

第二步 : 激活虚拟环境 source activate pytorch2caffe

第三步 : 在该虚拟环境下安装对应版本的pytorch和torchvision : conda install pytorch=0.2.0 torchvision=0.1.8 安装完成后conda list看一看有没有安装成功对应的版本

第四步 : 在该虚拟环境下编译caffe,官网指南链接 :
http://caffe.berkeleyvision.org/install_apt.html

第五步 : 编译配置caffe的python接口pycaffe,操作指南链接 :
https://www.cnblogs.com/lyyang/p/6573846.html

配置环境的时候因Git

广告一刻

为您即时展示最新活动产品广告消息,让您随时掌握产品活动新动态!