Caffe的draw_net的使用:draw_net.py绘制caffe net结构

1、安装pydot

$ sudo apt-get install python-pydot

2、安装graphviz

$ sudo apt-get install graphviz

3、生成

python draw_net.py ../examples/cifar10/cifar10_quick_train_test.prototxt cifa10.png

Caffe的draw_net的使用:draw_net.py绘制caffe net结构_第1张图片

/home/ubuntu/caffe/python/cifa10.png

 

python draw_net.py ../examples/cifar10/cifar10_quick_train_test.prototxt ../models/model_picture_s/cifa10.png
Drawing net to ../models/model_picture_s/cifa10.png

Caffe的draw_net的使用:draw_net.py绘制caffe net结构_第2张图片

结果:
Caffe的draw_net的使用:draw_net.py绘制caffe net结构_第3张图片

你可能感兴趣的:(机器学习,深度学习,算法学习,深度学习,caffe框架)