Caffe使用了Graphviz来绘制网络。本文叙述Windows环境。
Graphviz
Windows下,安装Graphviz Windows。
pydot
pydot
是python对graphviz的封装。调用Graphviz的dot.exe
。所以将含有dot.ext
的目录Graphviz/bin
添加到系统PATH
。
pip install pydot
draw_net
$CAFFE_ROOT/python
下,有该文件,用法如下:
python draw_net.py lenet_train_test.prototxt lenet.png --rankdir=TB
LeNet的结构如下: