使用GraphViz画caffe/torch的网络结构图
来源:互联网 发布:数控车g71编程实例详解 编辑:程序博客网 时间:2024/06/06 20:51
caffe的Python接口中有一个很不错的功能:画网络结构图,虽然画得并不好看,但可以给人一种直观的感受。
一、准备
首先caffe的python接口当然是必备的了,还没有生成python接口的同学可以参照我的上一篇博客来生成。
然后是需要安装protobuf的python接口,可以参照这篇博客进行安装,安装过程比较简单,就不赘述了。
安装GraphViz:http://www.graphviz.org/。
安装pydot,可以使用pip install pydot来安装,但最近这个功能好像不太稳定(GFW,你懂的),大家可以从https://github.com/nlhepler/pydot 下载,这里需要修改一下pydot/__init__.py文件中,第596行:
path = os.path.join(os.environ['PROGRAMFILES'], 'ATT', 'GraphViz', 'bin')
改为:
path = r"C:\Program Files (x86)\Graphviz2.37\bin"
同样,下边的那个path也改为r”C:\Program Files (x86)\Graphviz2.37\bin”,如果你将GraphViz安装在别的位置,那这个目录也要跟着改变。
二、生成网络结构图
把环境配置好之后,就可以在命令行中输入
python draw_net.py ..\examples\mnist\lenet_train_test.prototxt 1.bmp
就可以看到在当前目录下生成了一张图片:
torch reference:
Let's have a look:
-- some handy models are defined in optnet.models-- like alexnet, googlenet, vgg and resnetmodels = require 'optnet.models'modelname = 'googlenet'net, input = models[modelname]()generateGraph = require 'optnet.graphgen'-- visual properties of the generated graph-- follows graphviz attributesgraphOpts = {displayProps = {shape='ellipse',fontsize=14, style='solid'},nodeData = function(oldData, tensor) return oldData .. '\n' .. 'Size: '.. tensor:numel()end}g = generateGraph(net, input, graphOpts)graph.dot(g,modelname,modelname)
This generates the following graph:
https://github.com/DmitryUlyanov/optimize-net
DmitryUlyanov/optimize-net
- 使用GraphViz画caffe/torch的网络结构图
- 使用GraphViz画caffe/torch的网络结构图
- 使用GraphViz画caffe的网络结构图
- 使用GraphViz画caffe的网络结构图
- 使用GraphViz画caffe的网络结构图
- 使用GraphViz画caffe的网络结构图
- Caffe 使用Python绘制网络结构图是遇到的错误
- 利用Graphviz 画结构图
- 利用Graphviz 画结构图
- 利用Graphviz 画结构图
- 利用Graphviz 画结构图
- caffe系列(如何画网络结构图)
- Caffe绘制网络结构图
- caffe绘制网络结构图
- caffe绘制网络结构图
- caffe网络结构图绘制
- 用 Graphviz 画 神经网络结构图
- Caffe下python环境的编译和使用draw_net.py绘制lenet网络结构图
- PHP基础知识(小要点)
- commons-fileUpload组件解决文件上传乱码问题
- Cesium Appearance API
- 转载--java中遍历实体类属性和类型,属性值
- &和&& 的区别
- 使用GraphViz画caffe/torch的网络结构图
- 学习数据库必须掌握的54条SQL查询语句
- ScrollView ViewPager ListView三者共存
- 2017校招总结
- javaWeb入门<2>javaWeb中使用绝对路径与相对路径小结
- 关于宏的一些问题
- android添加常驻图标到状态栏
- mongodb 3.2.5安装过程详细记录
- 实战教你如何在短时间打造一个男性流量站