将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

Torch安装教程
Caffe安装教程
把Caffe转换为Torch

将path改变成自己的path
把 

.prototxt

 和 

.caffemodel

 放在同一个文件夹下.
会生成

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

确保

vgg16_torch.py

 和 

vgg16_torch.pth

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

vgg16_torch

意味着从

vgg16_torch.py

导入模型结构。

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()
...

参考资源

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

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/