文档章节

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

earnpls
 earnpls
发布于 2017/07/18 21:38
字数 410
阅读 125
收藏 0
点赞 0
评论 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/

© 著作权归作者所有

共有 人打赏支持
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

机器学习者必知的5种深度学习框架

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

雷锋字幕组 ⋅ 05/03 ⋅ 0

深度概率编程语言 Pyro 0.2 发布,支持 PyTorch 0.4

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

王练 ⋅ 04/27 ⋅ 0

PyTorch 重大更新,0.4.0 版本支持 Windows 系统

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

孔令双 ⋅ 04/25 ⋅ 0

一文读懂PyTorch张量基础(附代码)

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

技术小能手 ⋅ 06/13 ⋅ 0

Caffe2现已并入PyTorch,图像处理的孩子有福了!

源 | 小象 文 | 小象君 近日,GitHub发布一则通知,深度学习框架Caffe现已将代码并入了PyTorch。 Caffe(卷积神经网络框架),全称Convolutional Architecture for Fast Feature Embedding。...

机器学习算法与自然语言处理 ⋅ 04/15 ⋅ 0

PyTorch 重磅更新,不只是支持 Windows

翻译 | 林椿眄 出品 | AI 科技大本营(公众号ID:rgznai100) 这次版本的主要更新一些性能的优化,包括权衡内存计算,提供 Windows 支持,24个基础分布,变量及数据类型,零维张量,张量变量...

AI科技大本营 ⋅ 04/26 ⋅ 0

业界 | 无缝整合PyTorch 0.4与Caffe2,PyTorch 1.0即将问世

  选自Facebook Research   作者:Bill Jia   机器之心编译   参与:思源、晓坤      在 F8 的第二天中,Facebook 正式宣布 PyTorch1.0 即将与大家见面,这是继一周前发布 0.4....

机器之心 ⋅ 05/03 ⋅ 0

Github代码实践:Pytorch实现的语义分割器

雷锋网按:本文为雷锋字幕组编译的Github项目,原标题A Pytorch Implementation of Detectron,作者为 roytseng-tw。 翻译 | 杨婉迪 陈明霏 整理 | 凡江 使用Detectron预训练权重输出 e2emas...

雷锋字幕组 ⋅ 05/03 ⋅ 0

PyTorch 1.0 正式公开,Caffe2并入PyTorch实现AI研究和生产一条龙

今天,Facebook正式公布PyTorch 1.0,这是将基于Python的PyTorch与Caffe2合并的一个新版本的框架,让开发者可以无缝地将AI模型从研究转到生产,而无需处理迁移 “现在,你只需要使用PyTorch...

技术小能手 ⋅ 05/03 ⋅ 0

没有更多内容

加载失败,请刷新页面

加载更多

下一页

熊掌号收录比例对于网站原创数据排名的影响[图]

从去年下半年开始,我在写博客了,因为我觉得业余写写博客也还是很不错的,但是从2017年下半年开始,百度已经推出了原创保护功能和熊掌号平台,为此,我也提交了不少以前的老数据,而这些历史...

原创小博客 ⋅ 今天 ⋅ 0

LVM讲解、磁盘故障小案例

LVM LVM就是动态卷管理,可以将多个硬盘和硬盘分区做成一个逻辑卷,并把这个逻辑卷作为一个整体来统一管理,动态对分区进行扩缩空间大小,安全快捷方便管理。 1.新建分区,更改类型为8e 即L...

蛋黄Yolks ⋅ 今天 ⋅ 0

Hadoop Yarn调度器的选择和使用

一、引言 Yarn在Hadoop的生态系统中担任了资源管理和任务调度的角色。在讨论其构造器之前先简单了解一下Yarn的架构。 上图是Yarn的基本架构,其中ResourceManager是整个架构的核心组件,它负...

p柯西 ⋅ 今天 ⋅ 0

uWSGI + Django @ Ubuntu

创建 Django App Project 创建后, 可以看到路径下有一个wsgi.py的问题 uWSGI运行 直接命令行运行 利用如下命令, 可直接访问 uwsgi --http :8080 --wsgi-file dj/wsgi.py 配置文件 & 运行 [u...

袁祾 ⋅ 今天 ⋅ 0

JVM堆的理解

在JVM中,我们经常提到的就是堆了,堆确实很重要,其实,除了堆之外,还有几个重要的模块,看下图: 大 多数情况下,我们并不需要关心JVM的底层,但是如果了解它的话,对于我们系统调优是非常...

不羁之后 ⋅ 昨天 ⋅ 0

推荐:并发情况下:Java HashMap 形成死循环的原因

在淘宝内网里看到同事发了贴说了一个CPU被100%的线上故障,并且这个事发生了很多次,原因是在Java语言在并发情况下使用HashMap造成Race Condition,从而导致死循环。这个事情我4、5年前也经历...

码代码的小司机 ⋅ 昨天 ⋅ 2

聊聊spring cloud gateway的RetryGatewayFilter

序 本文主要研究一下spring cloud gateway的RetryGatewayFilter GatewayAutoConfiguration spring-cloud-gateway-core-2.0.0.RC2-sources.jar!/org/springframework/cloud/gateway/config/G......

go4it ⋅ 昨天 ⋅ 0

创建新用户和授予MySQL中的权限教程

导读 MySQL是一个开源数据库管理软件,可帮助用户存储,组织和以后检索数据。 它有多种选项来授予特定用户在表和数据库中的细微的权限 - 本教程将简要介绍一些选项。 如何创建新用户 在MySQL...

问题终结者 ⋅ 昨天 ⋅ 0

android -------- 颜色的半透明效果配置

最近有朋友问我 Android 背景颜色的半透明效果配置,我网上看资料,总结了一下, 开发中也是常常遇到的,所以来写篇博客 常用的颜色值格式有: RGB ARGB RRGGBB AARRGGBB 这4种 透明度 透明度...

切切歆语 ⋅ 昨天 ⋅ 0

CentOS开机启动subversion

建立自启动脚本: vim /etc/init.d/subversion 输入如下内容: #!/bin/bash## subversion startup script for the server## chkconfig: 2345 90 10# description: start the subve......

随风而飘 ⋅ 昨天 ⋅ 0

没有更多内容

加载失败,请刷新页面

加载更多

下一页

返回顶部
顶部