Caffe: 用Python绘制网络结构

来源:互联网 发布:tk域名注册不了 编辑:程序博客网 时间:2024/05/26 12:02

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的结构如下:

LeNet

0 0