编译caffe的Python接口
$ make pycaffe
装各种依赖
$ pip install pydot
$ sudo apt-get install graphviz
可视化模型
draw_net.py执行的时候带三个参数
第一个参数:网络模型的prototxt文件
第二个参数:保存的图片路径及名字
第二个参数:--rankdir=x , x 有四种选项,分别是LR, RL, TB, BT 。用来表示网络的方向,分别是从左到右,从右到左,从上到小,从下到上。默认为LR。
eg:
绘制googlenet模型
$ cd $(CAFFE_ROOT)/python/
$ ./draw_net.py ../models/bvlc_googlenet/deploy.prototxt bvlc_googlenet.caffemodel.jpg
$ make pycaffe
装各种依赖
$ pip install pydot
$ sudo apt-get install graphviz
可视化模型
draw_net.py执行的时候带三个参数
第一个参数:网络模型的prototxt文件
第二个参数:保存的图片路径及名字
第二个参数:--rankdir=x , x 有四种选项,分别是LR, RL, TB, BT 。用来表示网络的方向,分别是从左到右,从右到左,从上到小,从下到上。默认为LR。
eg:
绘制googlenet模型
$ cd $(CAFFE_ROOT)/python/
$ ./draw_net.py ../models/bvlc_googlenet/deploy.prototxt bvlc_googlenet.caffemodel.jpg