Linux下conda安装caffe（超简单），pb转caffe

最近碰到了需要将tensorflow的pb模型转为caffe的prototxt格式的问题，参考了

keras/tensorflow1.x模型转换为caffe推理模型 - 知乎

mmdnn tensorflow 转 caffe

发现除了安装mmdnn以外，还依赖caffe，配置caffe环境会遇到各种坑，因为我不需要caffe训练模型，只是为了转换，因此只需要装caffe-cpu版本即可，最后经过不断尝试，在百度的AI Studio上安装上了caffe，这里记录一下caffe的安装：

Anaconda的安装就不赘述了，百度的AI Studio上已经有conda了，因此直接从caffe的安装开始：

1.先新建一个python3.5环境

conda create -n caffe-py3.5 python=3.5 -c defaults

输入 y 进行安装

2.激活环境

安装完毕后，激活进入环境

source activate caffe-py3.5

3.安装caffe-cpu版本或caffe-gpu版本

conda install -c defaults caffe
conda install -c defaults caffe-gpu

4.测试安装是否成功：

python3
import caffe

无报错说明安装成功，退出python

exit()

5.安装tensorflow：

conda install -c defaults tensorflow==1.15.0

6.安装mmdnn

pip install mmdnn

6. 我们以tf官网下载的mobilenetv1模型为例，转换前先使用netron查看pb模型输入和输出节点name，input shape：

记住name以及input shape

7.进行pb ----> caffe转换，命令行输入：

mmconvert -sf tensorflow -iw mobilenet_v1.pb --inNodeName input --inputShape 192,192,3 --dstNodeName MobilenetV1/Predictions/Reshape_1 -df caffe -om tf_mobilenet

8.转换结果：

成功转换为caffe模型：

总结:

caffe还是在Linux下用conda安装最简单方便！

实测，python3.6、python3.5都可以按照上述方法安装caffe

===================================================================

Reference:

1.Anaconda安装caffe（超简单）_abcd740181246的博客-CSDN博客_anaconda安装caffe

2.安装教程：使用Anaconda创建caffe和tensorflow共存环境_PRIMEZPY的博客-CSDN博客

===================================================================

Linux下conda安装caffe（超简单），pb转caffe相关推荐

Kali Linux 2021.2在VMware和VirtualBox安装教程超简单步骤详细讲解
Kali Linux 2021.2在VMware和VirtualBox安装教程超简单步骤详细讲解一. 资源下载及工作站安装二. 使用VMware平台安装三. 使用VirtualBox平台安装 ...
linux 下mysql的管理,Linux下 MySQL安装和基本管理
•安装linux一些必须的包: 1.yum install gcc gcc-c++ gcc-g77 autoconf automake zlib* fiex* libxml* ncurses-deve ...
linux tomcat 安装教程,关于Linux下配置安装Tomcat8的步骤
由于Centos6即将停止更新,本次就使用Centos7.5 X64位系统来演示,Tomcat版本就使用8.5,如果想使用9.0的版本也可以去官网进行下载,步骤都大同小异. 关于Tomcat9的新特性 ...
mac安装telnet 超简单复制telnet文件即可
mac安装telnet 超简单复制telnet文件即可: telnet 下载链接: https://github.com/cuifeiran/macos-telnet 将telnet可执行文件扔到m ...
Linux 下如何安装软件？
http://zhidao.baidu.com/link?url=OkQCOZtVMXhasC8x9zFTZOumsFKf0WW25Ckr2wBF1xO08EsjrBpnMaTBlIAUYdxZ408 ...
linux下python安装opencv库_Linux下怎么配置python和opencv
匿名用户 1级 2017-06-20 回答以下说明在Linux下Python和OpenCV结合安装的过程,Python要使用OpenCV模块,则必须导入OpenCV提供的包,所以要提供Python支 ...
7z(p7zip)压缩软件在Linux下的安装和使用
为什么80%的码农都做不了架构师?>>> Evernote Export 7z(p7zip)压缩软件在Linux下的安装和使用 1) 简介 7z,全称7-Zip, 是一款开源软 ...
Linux 下 Redis 安装教程
Linux 下 Redis 安装教程事先准备下载 Redis 安装 Redis 设置 Redis 开机自启动远程访问 Redis RDM 远程访问 Redis 笔者的运行环境: CentOS S ...
linux下成功安装ffmpeg（亲测有效）
linux下成功安装ffmpeg( 亲测有效 ) 一.下载二.安装步骤 1.安装yasm 2.安装ffmpeg 总结一.下载 ffmpeg 官网下载: http://ffmpeg.org/down ...

Linux下conda安装caffe（超简单），pb转caffe

Linux下conda安装caffe（超简单），pb转caffe相关推荐

最新文章

热门文章