1. 程式人生 > >python繪製caffe例項的網路模型

python繪製caffe例項的網路模型

在所有目錄與檔案均是基於caffe根目錄!!!!!
python/draw_net.py可以將網路模型由prototxt變成一張圖片。

在繪製之前,先安裝兩個庫:::

1、安裝GraphViz

sudo apt-get install GraphViz

Graphviz的是一款圖形繪製工具,用來被python程式呼叫繪製圖片。

2、安裝pydot

sudo pip install pydot

pydot是python的支援畫圖的庫

編譯pycaffe

在繪製之前,還要執行下述語句編譯_caffe.cpp

 make pycaffe

否則會報如下圖所示錯誤:::
這裡寫圖片描述

完成上述三步後,就可以呼叫指令碼來繪製圖片了

draw_net.py執行的時候帶三個引數

第一個引數:網路模型的prototxt檔案

第二個引數:儲存的圖片路徑及名字

第二個引數:–rankdir=x , x 有四種選項,分別是LR, RL, TB, BT 。用來表示網路的方向,分別是從左到右,從右到左,從上到小,從下到上。預設為LR。

繪製Lenet模型

python python/draw_net.py examples/mnist/lenet_train_test.prototxt ./sc.jpg --rankdir=BT

這裡寫圖片描述

繪製cifar10的模型

python python/draw_net.py
examples/cifar10/cifar10_full_train_test.prototxt cifar10_net.jpg --rankdir=BT

這裡寫圖片描述