《深度學習 實戰caffe》 繪製網路模型

2021-08-20 14:43:47 字數 1042 閱讀 5805

python/draw_net.py, 這個檔案,就是用來繪製網路模型的。也就是將網路模型由prototxt變成一張。

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

1、安裝graphviz

# 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 。用來表示網路的方向,分別是從左到右,從右到左,從上到小,從下到上。預設為lr。

繪製lenet-5模型:

python draw_net.py ../models/bvlc_reference_caffenet/train_val.prototxt caffenet.png
例:繪製lenet模型

例:繪製cifar10的模型

caffe繪製網路

計算均值命令 sudo home tyd caffe build tools compute image mean home tyd caffe examples mnist mnist train lmdb home tyd caffe case mean.binaryproto 繪製網路 1.安...

深度學習實戰Caffe 一 Caffe環境安裝

安裝vim git 等必備工具 sudo apt get install vim git 安裝boost sudo apt get install no install recommends libboost all dev安裝atlas開發包 sudo apt get install libatl...

深度學習 21天實戰Caffe

幫助 內容簡介 深度學習 21天實戰caffe 是一本深度學習入門讀物。以目前已經大量用於線上系統的深度學習框架caffe為例,由淺入深,從 caffe 的配置 部署 使用開始學習,通過閱讀 caffe 原始碼理解其精髓,加強對深度學習理論的理解,最終達到熟練運用 caffe 解決實際問題的目的。和...