可视化神经网络结构图

安装依赖库

sudo apt-get install graphviz #这里使用apt-get安装，不是使用pip安装
pip install  pydot

使用draw_net.py绘制结构图

cd caffe/python  #进入draw_net.py目录下

python draw_net.py --rankdir TB ../examples/mnist/lenet.prototxt ../examples/mnist/lenet_structure.jpg

＃rankdir选项确定了输出图片的摆放位置，有四种选项：LR, RL, TB, BT ，分别是从左到右，从右到左，从上到下，从下到上；默认为ＬＲ

参考文献

[1] https://blog.csdn.net/HMH2_YY/article/details/78430875