将Caffe Model转化为Pytorch Model以及Torch Model

将Caffe转化为Pytorch

1、将Caffe Model转化为Pytorch Model

2、将Caffe Model转化为Torch Model

3、将Torch Model转化为Pytorch Model

我已经测试了vgg16，它表现在分类任务。但我不能保证它在其他任务上执行得很好（如对象检测和语义分割）。您可以尝试并根据bug信息修改代码。如果有你的咖啡模型新的组件，你应该在代码中添加相应的部分。Github地址：https://github.com/fanq15/caffe_to_torch_to_pytorch

安装Torch和Caffe

1、Torch安装教程

2、Caffe安装教程

把Caffe转换为Torch

1、将path改变成自己的path

2、把

.prototxt

和

.caffemodel

放在同一个文件夹下.

3、会生成

vgg16_torch.t7

文件夹.

th caffemodel_to_t7.lua

把torch转换为pytorch

python convert_torch.py -m vgg16_torch.t7

将会创建2个文件

vgg16_torch.py

vgg16_torch.pth

在python中加载

.pth model

1、确保

vgg16_torch.py

和

vgg16_torch.pth

文件在python工作区的同一个文件夹下

2、导入

vgg16_torch

意味着从

vgg16_torch.py

导入模型结构。

3、

model.load_state_dict

代表从vgg16_torch.pth模型中家在权重

import vgg16_torch

model = vgg16_torch.vgg16_torch
model.load_state_dict(torch.load('vgg16_torch.pth'))
model.eval()
...

参考资源

1、Caffe转换为Torch参考 https://github.com/jcjohnson/pytorch-vgg

2、Torch转Pytorch参考 https://github.com/clcarwin/convert_torch_to_pytorch

在这里，我整理发布了Pytorch中文文档，方便大家查询使用，同时也准备了中文论坛，欢迎大家学习交流！

Pytorch中文文档

Pytorch中文论坛

Pytorch中文文档已经发布，完美翻译，更加方便大家浏览：

Torch中文网：https://ptorch.com/

Pytorch中文网：https://ptorch.com/

Pytorch中文文档：https://ptorch.com/docs/1/