1. 程式人生 > >Caffe學習系列(1): 繪製網路模型

Caffe學習系列(1): 繪製網路模型

1、安裝必要的庫。

$ sudo apt-get install GraphViz
$ sudo pip install pydot

2、繪製網路
引數說明:
draw_net.py執行的時候帶三個引數
第一個引數:網路模型的prototxt檔案
第二個引數:儲存的圖片路徑及名字
第二個引數:–rankdir=x , x 有四種選項,分別是LR, RL, TB, BT 。用來表示網路的方向,分別是從左到右,從右到左,從上到小,從下到上。預設為LR。
例子:

$sudo python python/draw_net.py examples/mnist/lenet_train_test.prototxt net.png --rankdir=BT


這裡寫圖片描述

3、繪製自己的網路模型
未完待續。。。