《深度學習——實戰caffe》——繪制網絡模型

python/draw_net.py, 這個文件，就是用來繪制網絡模型的。也就是將網絡模型由prototxt變成一張圖片。

在繪制之前，需要先安裝兩個庫

１、安裝ＧraphViz

# sudo apt-get install GraphViz

注意，這里用的是apt-get來安裝，而不是pip.

2 、安裝pydot

# sudo pip install pydot

用的是pip來安裝，而不是apt-get

安裝好了，就可以調用腳本來繪制圖片了

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

第一個參數：網絡模型的prototxt文件

第二個參數：保存的圖片路徑及名字

第二個參數：--rankdir=x , x 有四種選項，分別是LR, RL, TB, BT 。用來表示網絡的方向，分別是從左到右，從右到左，從上到小，從下到上。默認為ＬＲ。

繪制LeNet-5模型：

python draw_net.py ../models/bvlc_reference_caffenet/train_val.prototxt caffenet.png

例：繪制Lenet模型

# sudo python python/draw_net.py examples/mnist/lenet_train_test.prototxt netImage/lenet.png --rankdir=BT

例：繪制cifar10的模型

#  sudo python python/draw_net.py examples/cifar10/cifar10_full_train_test.prototxt netImage/cifar10.png --rankdir=BT

本文鏈接：https://blog.csdn.net/qq_33144323/article/details/80638852

智能推薦

caffe入門學習（5）：繪制網絡結構圖

網絡配置都是在prototxt文件中看起來不直觀，如果能直接看到各層的連接圖就了，Caffe提供了這樣的工具。本文就是描述我用draw_net.py繪制網絡結構圖的踏坑過程。 1、安裝graphviz 繪制圖形，首先我們需要一個圖形包。 1）下載安裝包 https://graphviz.gitlab.io/_pages/Download/Download_windows.html 2）安裝程序執...

《深度學習——實戰caffe》——初識數據可視化

首先將caffe的根目錄作為當前目錄，然后加載caffe程序自帶的小貓圖片，并顯示。圖片大小為360x480，三通道打開examples/net_surgery/conv.prototxt文件，修改兩個地方一是將input_shape由原來的是（1，1，100，100）修改為(1,3,100,100),即由單通道灰度圖變為三通道彩色圖。二是將過濾器個數(num_output)由3修改為16...

《深度學習——實戰caffe》——caffemodel可視化

通過前面的學習，我們已經能夠正常訓練各種數據了。設置好solver.prototxt后，我們可以把訓練好的模型保存起來，如lenet_iter_10000.caffemodel。訓練多少次就自動保存一下，這個是通過snapshot進行設置的，保存文件的路徑及文件名前綴是由snapshot_prefix來設定的。這個文件里面存放的就是各層的參數，即net.params，里面沒有數據(net.blob...

深度學習Caffe實戰筆記（1）環境搭建

（1）環境搭建從知道深度學習開始，就一直想學習使用caffe，礙于各種事情一直沒有如愿，這幾天終于找了個時間搞了一下，打算把學習的過程整理成筆記，包括環境搭建、跑車牌識別數據，跑mnist數據，用Alexnet跑自己的數據，用Siamese網絡跑mnist數據，用Siamese網絡跑自己的數據以及如何調整網絡結構等等。。。。后續我會慢慢更新，筆記的主要內容是如何使用caffe，主要側重于實戰，基...

深度學習21天實戰caffe學習筆記《6 : Caffe代碼梳理》

Caffe代碼梳理 1、caffe目錄結構 2、caffe閱讀路線： src/caffe/proto/caffe.proto 了解基本數據結構內存對象和磁盤文件的一一映射，主要由ProtoBuffer工具完成； include頭文件 ...

《深度學習——實戰caffe》——繪制網絡模型

智能推薦

猜你喜歡