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

将Caffe转化为Pytorch

1. 1、将Caffe Model转化为Pytorch Model
2. 2、将Caffe Model转化为Torch Model
3. 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/