Caffe net结构可视化

简介

net可视化主要是以流程图的形式把caffe中使用的模型(每一个layer以及layer和layer之间的链接)展示出来,这样更加直观和具体。caffe的python接口下面已经提供了可视net 的功能,就是’draw_net.py’文件。python

编译pycaffe

编译caffe时必须编译了caffe的python的wrapper,命令以下,这样才能正常使用caffe下的python接口web

make pycaffe

安装pydot

python的pydot模块是Graphviz的dot语言的python接口,这里主要使用Graphviz绘图,固然要安装pydot
网址:https://pypi.python.org/pypi/pydot
使用下面命令安装。bash

source ENV/bin/activate
pip install pydot

安装Graphviz

Graphviz是大名鼎鼎的贝尔实验室的几位牛人开发的一个画图工具。它的理念和通常的“所见即所得”的画图工具不同,是“所想即所得”。Graphviz提供了dot语言来编写绘图脚本。
网址:http://www.graphviz.org/
能够根据本身的系统下载相应的deb文件安装。
下载源码根据INSTALL编译安装,执行下面命令进行编译安装。网络

cd graphviz_dir
./configure
make 
make check
sudo make insall
make installcheck

绘制

绘制mnist的LeNet网络层结构app

cd caffe_root/python
python draw_net.py ../examples/mnist/lenet_train_test.prototxt lenet.png

结果
LeNetsvg

本文同步分享在 博客“无比机智的永哥”(CSDN)。
若有侵权,请联系 support@oschina.cn 删除。
本文参与“OSC源创计划”,欢迎正在阅读的你也加入,一块儿分享。工具

相关文章
相关标签/搜索