文档章节

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

earnpls
 earnpls
发布于 2017/07/18 21:38
字数 410
阅读 169
收藏 0

将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/

本文转载自:https://ptorch.com/news/26.html

共有 人打赏支持
earnpls
粉丝 5
博文 26
码字总数 74
作品 0
昌平
程序员
整合PyTorch 0.4和Caffe 2,PyTorch 1.0能挑战TensorFlow吗?

 译者 | 梁红丽 编辑 | Mavis 出品 | AI科技大本营(公众号ID:rgznai100) 【AI 科技大本营导读】5月2日,在加利福尼亚州举办的年度开发者 F8 大会上,Facebook 正式推出 PyTorch 1.0 ...

dqcfkyqdxym3f8rb0
05/04
0
0
机器学习者必知的5种深度学习框架

雷锋网按:本文为雷锋字幕组编译的技术博客,原标题The 5 Deep Learning Frameworks Every Serious Machine Learner Should Be Familiar With,作者为James Le。 翻译 | 杨恕权 张晓雪 陈明霏...

雷锋字幕组
05/03
0
0
深度概率编程语言 Pyro 0.2 发布,支持 PyTorch 0.4

Pyro 是 Uber AI 实验室开源的一款深度概率编程语言(PPL),基于 Python 与 PyTorch 之上,专注于变分推理,同时支持可组合推理算法。 Pyro 能够实现灵活而富有表现力的深度概率建模,将现代...

王练
04/27
0
0
PyTorch 重大更新,0.4.0 版本支持 Windows 系统

雷锋网(公众号:雷锋网) AI 研习社最新消息,PyTorch 官方发布 0.4.0 版本,该版本的 PyTorch 有多项重大更新,其中最重要的改进是支持 Window 系统。 2017 年初,Facebook 在机器学习和科学...

孔令双
04/25
0
0
一文读懂PyTorch张量基础(附代码)

本文介绍了PyTorch中的Tensor类,它类似于Numpy中的ndarray,它构成了在PyTorch中构建神经网络的基础。 我们已经知道张量到底是什么了,并且知道如何用Numpy的ndarray来表示它们,现在我们看...

技术小能手
06/13
0
0

没有更多内容

加载失败,请刷新页面

加载更多

TypeScript基础入门之高级类型的可辨识联合(Discriminated Unions)

转发 TypeScript基础入门之高级类型的可辨识联合(Discriminated Unions) 高级类型 可辨识联合(Discriminated Unions) 你可以合并单例类型,联合类型,类型保护和类型别名来创建一个叫做 ...

durban
11分钟前
1
0
画图工具

目标是不要让我自己摆放,那样会打断我的思路。要自己会摆放,我写字就好。 纠结对齐的我在也不会因为一个像素的问题小心翼翼的移动鼠标了。因为它对不齐你也管不了。 Graphviz https://www...

郭大鹏
11分钟前
1
0
在Visual Studio中使用GitHub(使用篇)

上一篇中已经介绍了如何在windows下安装和配置Git,并且详细说明了如何和GitHub连接。 这里就详细的说明下如何在Visual Studio中使用GitHub。这里也是一个简单的Git教程。 注:非常非常简单的...

Jokeny
12分钟前
1
0
tomcat 在liunx系统中shutdown后进程仍然存在解决办法

这个在windows下没有碰到过,因为此前跑Tomcat都是以服务而不是命令脚本的形式跑的,而且已经换了一个项目,所以暂时不考察windows下是否存在该问题。 难道是Tomcat版本问题?或者用带内存泄...

乱舞
13分钟前
1
0
区块链10年了,还未落地,它失败了吗?

几乎每个人,甚至是对通证持怀疑态度的人,都对区块链的技术有积极的看法,因为它有可能改变世界。然而,区块链技术问世已经10年了,我们仍然没有真正的用上区块链技术。 有人说,比特币是建...

linux-tao
17分钟前
1
0

没有更多内容

加载失败,请刷新页面

加载更多

返回顶部
顶部