Caffe 网络结构可视化
2019独角兽企业重金招聘Python工程师标准>>>
可以利用python接口实现网络结构的可视化,便于直观理解。
最近安装了tensorflow,发现caffe的draw_net脚本不好用了(AttributeError: ‘google.protobuf.pyext._message.RepeatedScalarConta’ object has no attribute ‘_values’),原因是二者的protobuf版本不兼容。caffe不支持高于3.0.0版本的protobuf.
解决方法:
卸载当前protobuf。(pip uninstall protobuf)
再指定安装低版本的,这里用的是2.5.0(pip install protobuf==2.5.0)
可以用virtualenv创建独立的Python环境
在线网页绘制caffe网络结构图的方法,其实更简洁方便:
http://ethereon.github.io/netscope/#/editor
注:主要参考《21天实战caffe》
1.准备Python环境
1.1安装Python
sudo apt-get update
sudo apt-get install python-dev
- 1
- 2
1.2安装Python包管理器:pip
这里建议通过源码安装,如果通过命令 sudo apt-get install python-pip 安装,可能会出现类似下面的错误,主要原因是上述方式安装的pip版本较老。
Exception:
Traceback (most recent call last):File "/usr/lib/python2.7/dist-packages/pip/basecommand.py", line 122, in mainstatus = self.run(options, args)File "/usr/lib/python2.7/dist-packages/pip/commands/install.py", line 278, in runrequirement_set.prepare_files(finder, force_root_egg_info=self.bundle, bundle=self.bundle)File "/usr/lib/python2.7/dist-packages/pip/req.py", line 1260, in prepare_files)[0]
IndexError: list index out of range
- 1
- 2
- 3
- 4
- 5
- 6
- 7
- 8
- 9
源码安装pip方式如下(参考:http://www.cnblogs.com/ajianbeyourself/p/4214398.html):
wget https://bootstrap.pypa.io/get-pip.py --no-check-certificate
sudo python get-pip.py
- 1
- 2
或者直接更新pip:
pip install -U pip
- 1
1.3通过pip安装依赖包
这里caffe已经列出了依赖包及版本号,在caffe源码根目录下的 /python/requirement.txt 中:
Cython>=0.19.2
numpy>=1.7.1
scipy>=0.13.2
scikit-image>=0.9.3
matplotlib>=1.3.1
ipython>=3.0.0
h5py>=2.2.0
leveldb>=0.191
networkx>=1.8.1
nose>=1.3.0
pandas>=0.12.0
python-dateutil>=1.4,<2
protobuf>=2.5.0
python-gflags>=2.0
pyyaml>=3.10
Pillow>=2.3.0
six>=1.1.0
sudo apt-get install python-numpy #之前以为和下面pip安装的重复了,但是不安装编译时会报numpy的错误。
sudo pip install -r ${CAFFE_ROOT}/python/requirement.txt
- 1
- 2
2.编译 pycaffe
cd ${CAFFE_ROOT}
make clean
make
make pycaffe
- 1
- 2
- 3
- 4
3.绘制网络结构图
cd ${CAFFE_ROOT}
python ./python/draw_net.py ./models/bvlc_reference_caffenet/train_val.prototxt ~/Desktop/caffenet.png
- 1
- 2
- 3
4.“dot” not found in path 的问题
2016.12.01添加:
先安装graphviz否则会出现类似:“dot” not found in path 的问题
安装graphviz不要用pip install安装,否则还是会找不到可执行程序
安装:sudo apt-get insall graphviz
然后安装pydot:pip install pydot
其中pyparsing会自动安装
摘自:pydot 安装 用于caffe画图
转载于:https://my.oschina.net/lilinzero/blog/884699
Caffe 网络结构可视化相关推荐
- FCN Caffe:可视化featureMaps和Weights(C++)、获取FCN结果
为何不使用C++版本FCN获取最后的分割掩模像素块集合,何必要使用python呢!因此需要获取网络最后层的所有featureMaps,featureMaps的结果直接对应了segmentation的最 ...
- Pytorch的网络结构可视化(tensorboardX)(详细)
20210610 if config.test is True:model = load_test_model(model, config)print(model) 打印网络结构 版权声明:本文为博主 ...
- Pytorch 网络结构可视化
点击上方"小白学视觉",选择加"星标"或"置顶" 重磅干货,第一时间送达 本文转自:深度学习这件小事 安装 可以通过以下的命令进行安装 co ...
- Pytorch网络结构可视化
现在用的这个: net = load_model(net, args.trained_model, args.cpu) for name, param in net.named_parameters( ...
- Pytorch的网络结构可视化:Netron与TensorBoardX
Pytorch的网络结构可视化:Netron与TensorBoardX Pytorch的网络结构可视化:Netron Pytorch的网络结构可视化:TensorBoardX 1.TensorBoar ...
- 卷积神经网络结构可视化工具PlotNeuralNet
卷积神经网络结构可视化 简介 本文介绍的工具是针对卷积神经网络示意图可视化的,不包括算图.(示意图一般出现在论文中)常见的卷积神经网络示意图绘制工具不少,常用的主要有NN SVG.ConvNetDra ...
- Netron神经网络结构可视化只显示权重没有箭头,已解决
Netron神经网络结构可视化只显示权重没有箭头,已解决 好不容易写好了网络,保存成了.h5格式,放进Netron里作图发现我的图黑黑丑丑的而且只显示权重,如下图: 可是看别人画得花花绿绿的还漂亮,为 ...
- Caffe网络结构实现
对于神经网络实现手写数字识别(MNIST)网络结构通过在线可视化工具查看和修改: (http://ethereon.github.io/netscope/#/editor ) 一.卷积层(Convol ...
- 【caffe】可视化网络的训练过程的loss和accuracy
转载自https://blog.csdn.net/chen1234520nnn/article/details/102780652 前言:我们在训练网络的时候,网络训练完毕后,需要可视化训练过程中lo ...
最新文章
- Caffe 中关于 LetNet-5 网络的定义文件 lenet.prototxt 解析
- 应用程序的数据库从Sql Server迁移到Oracle
- MSSQL分组取后每一组的最新一条记录
- 【树莓派学习笔记】五、处理、自动重命名并另存为图片
- ubuntu安装字符集
- html-css样式表
- 去哪儿-09-city-list
- 理解Vue的计算属性
- 百度系无人车创业公司领骏科技完成新一轮融资
- JavaScript的Forms验证-Parsley.js
- 利用ESXI-Customizer制作定制化的ESXI
- Java中快速生成UUID
- 最著名的数学家一般也是最著名的力学家
- 2022年计算机视觉3大趋势
- 【ENSP模拟器】ENSP——VLAN的配置
- IDA遇到mojava,crash的情况
- 一个1-23二进制搭建方法
- CREATE TABLE 时的 SQL UNIQUE 约束
- KDD CUP 2009年数据集介绍及链接
- 东吴证券分析报告(0607)