文档章节

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

earnpls
 earnpls
发布于 2017/07/18 21:41
字数 410
阅读 203
收藏 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/

本文转载自: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
Keras vs PyTorch:谁是「第一」深度学习框架?

  选自Deepsense.ai   作者:Rafa Jakubanis、Piotr Migdal   机器之心编译   参与:路、李泽南、李亚洲      「第一个深度学习框架该怎么选」对于初学者而言一直是个头疼的问题...

机器之心
06/30
0
0
一文读懂PyTorch张量基础(附代码)

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

技术小能手
06/13
0
0
Github 最受欢迎的深度学习项目 TOP 20

必须做好一些是LINUX平台,甚至没有教材根本需要摸索学习的准备,如果有那个变成为在线服务的请第一时间通知我们 截止到 2017年 5月14日,Github 最受欢迎的深度学习项目 TOP 20 是哪些吧(统...

wotacid
2017/05/23
0
0
Caffe2现已并入PyTorch,图像处理的孩子有福了!

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

机器学习算法与自然语言处理
04/15
0
0
PyTorch 重磅更新,不只是支持 Windows

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

AI科技大本营
04/26
0
0
pyTorch版OpenNMT的学习笔记

前言 2017年1月18日Touch7的开发团队发布了pyTorch,pyTorch是一个python优先的深度学习框架,能够在GPU加速的基础上实现Tensor计算和动态神经网络。 是的,相较于G家以静态图为基础的tenso...

AlexTuring
2017/05/13
0
0

没有更多内容

加载失败,请刷新页面

加载更多

下一页

Tomcat内存设置

第一步骤: 1、路径是Tomcat目录下的/bin/catalina.bat文件 打开catalina.bat文件 加入下面这句代码: set JAVA_OPTS=%JAVA_OPTS% -server -XX:PermSize=128M -XX:MaxPermSize=512m 2、路径是...

一梦心草
12分钟前
0
0
IDEA 及 Gradle 使用总结

IDEA 及 Gradle 使用总结 自动编译组件 目前Android开发的主流开发工具是 Eclipse 和 IDEA 目前主流的自动化打包工具时 ant,maven,gradle。 maven工具中有自己的依赖仓库维护,很多开源支持...

舒文joven
13分钟前
0
0
lombok 引入后,测试类始终找不到get,set方法。

开发环境为idea,jdk1.7,maven3.5. 网上直接搜出来的方法有: 1、在setting里安装lombok的plugins; 2、如下图,勾选enable annocation processing选项 3、升级maven plugins插件 我尝试了以...

Kidult
14分钟前
0
0
Duang,HUAWEI DevEco IDE全面升级啦

想感受全新UI带来的视觉及交互体验、 HiKey970开发板调测、 HiAI API推荐和收藏、 深度AI模型分析等新功能, 体验高清晰度和流畅度的远程AI真机调测吗? 全新的UI设计 采用最优秀的视觉及交互...

华为终端开放实验室
23分钟前
1
0
阻止事件冒泡,阻止默认事件

1.event.stopPropagation()方法 这是阻止事件的冒泡方法,不让事件向documen上蔓延,但是默认事件任然会执行,当你掉用这个方法的时候,如果点击一个连接,这个连接仍然会被打开, 2.event....

闫亚亚
25分钟前
0
0
网络监控工具类

package com.guorentong.learn.organ.utils;import android.app.Activity;import android.content.Context;import android.content.Intent;import android.net.ConnectivityManage......

王先森oO
25分钟前
0
0
命令

sudo nginx -s reload 重启nginx sudo lsof -i -P | grep -i "listen" 查看端口占用

那个猴子
29分钟前
0
0
用scrapy-redis爬去新浪-以及把数据存储到

需求:爬取新浪网导航页(http://news.sina.com.cn/guide/)所有下所有大类、小类、小类里的子链接,以及子链接页面的新闻内容。 准备工作: a.安装redis(windows或者linux) b.安装Redis Des...

丁典
29分钟前
0
0
PHP常用函数篇

1.为什么要使用函数? 除了内建的PHP函数,我们可以创建我们自己的函数。 函数是可以在程序中重复使用的语句块。 使代码逻辑更清晰 避免过多的全局变量 封装后避免相同逻辑重复代码,只需调用...

天地有涯风有信_大海无量不见人
30分钟前
0
0
对List分组

在日常工作中会遇到这样的情景,我们需要对List按照List中对象的一个值进行分组。比如一个Human的List,我们要根据性别分组,传统的方法是做双层循环,逐个对比,今天我要介绍一种详单简单的...

珂jack
32分钟前
0
0

没有更多内容

加载失败,请刷新页面

加载更多

下一页

返回顶部
顶部