這是無意中發現的

原來caffe 有內建函數draw_net

讓使用者畫出網路圖

賈先生太強了~~

廢話不多說 開始吧

 

一.安裝

首先安裝必要項目 pydot 

sudo apt-get install python-pydot

 

建議不要用  pip install pydot

小編之前試過

但程式報錯  還要再 pip uninstall pydot 回來

 

二.使用

使用也超簡單的

python draw_net.py --rankdir XX  input_net_proto_file output_image_file

XX 可以填產生出的圖片位置

TB (Top-Bottom) 

RL (Right-Left）

LR (Left-Right) 

如果都不打 默認是LR

 

例子:

python draw_net.py --rankdir TB ../examples/mnist/lenet.prototxt lenet.png 

 

三.產生圖片

arrow
arrow
    文章標籤
    draw_net caffe教學 caffe 使用 畫出網路圖
    全站熱搜

    Darwin的AI天地 發表在 痞客邦 留言(0) 人氣()

    E-mail轉寄