将Caffe Model转化为Pytorch Model以及Torch Model
博客专区 > earnpls 的博客 > 博客详情
将Caffe Model转化为Pytorch Model以及Torch Model
earnpls 发表于5个月前
将Caffe Model转化为Pytorch Model以及Torch Model
  • 发表于 5个月前
  • 阅读 151
  • 收藏 0
  • 点赞 0
  • 评论 0

腾讯云 技术升级10大核心产品年终让利>>>   

将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. 1、Torch安装教程
  2. 2、Caffe安装教程

把Caffe转换为Torch

  1. 1、将path改变成自己的path
  2. 2、把 .prototxt 和 .caffemodel 放在同一个文件夹下.
  3. 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. 1、确保vgg16_torch.py 和 vgg16_torch.pth 文件在python工作区的同一个文件夹下
  2. 2、导入vgg16_torch意味着从vgg16_torch.py导入模型结构。
  3. 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. 1、Caffe转换为Torch参考 https://github.com/jcjohnson/pytorch-vgg

  2. 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/

标签: pytorch Torch7 Caffe
共有 人打赏支持
粉丝 5
博文 26
码字总数 74
×
earnpls
如果觉得我的文章对您有用,请随意打赏。您的支持将鼓励我继续创作!
* 金额(元)
¥1 ¥5 ¥10 ¥20 其他金额
打赏人
留言
* 支付类型
微信扫码支付
打赏金额:
已支付成功
打赏金额: