2019獨角獸企業重金招聘Python工程師標準>>>

可以利用python接口實現網絡結構的可視化，便于直觀理解。

最近安裝了tensorflow，發現caffe的draw_net腳本不好用了（AttributeError: ‘google.protobuf.pyext._message.RepeatedScalarConta’ object has no attribute ‘_values’），原因是二者的protobuf版本不兼容。caffe不支持高于3.0.0版本的protobuf.

解決方法：?
卸載當前protobuf。（pip uninstall protobuf）?
再指定安裝低版本的，這里用的是2.5.0（pip install protobuf==2.5.0）?
可以用virtualenv創建獨立的Python環境

在線網頁繪制caffe網絡結構圖的方法，其實更簡潔方便：?
http://ethereon.github.io/netscope/#/editor

注：主要參考《21天實戰caffe》

1.準備Python環境

1.1安裝Python

sudo apt-get update sudo apt-get install python-dev

• 1
• 2

1.2安裝Python包管理器：pip?
這里建議通過源碼安裝，如果通過命令 sudo apt-get install python-pip 安裝，可能會出現類似下面的錯誤，主要原因是上述方式安裝的pip版本較老。

Exception: Traceback (most recent call last):File "/usr/lib/python2.7/dist-packages/pip/basecommand.py", line 122, in mainstatus = self.run(options, args)File "/usr/lib/python2.7/dist-packages/pip/commands/install.py", line 278, in runrequirement_set.prepare_files(finder, force_root_egg_info=self.bundle, bundle=self.bundle)File "/usr/lib/python2.7/dist-packages/pip/req.py", line 1260, in prepare_files)[0] IndexError: list index out of range

• 1
• 2
• 3
• 4
• 5
• 6
• 7
• 8
• 9

源碼安裝pip方式如下（參考：http://www.cnblogs.com/ajianbeyourself/p/4214398.html）：

wget https://bootstrap.pypa.io/get-pip.py --no-check-certificate sudo python get-pip.py

• 1
• 2

或者直接更新pip：

pip install -U pip

• 1

1.3通過pip安裝依賴包?
這里caffe已經列出了依賴包及版本號，在caffe源碼根目錄下的 /python/requirement.txt 中：?
Cython>=0.19.2?
numpy>=1.7.1?
scipy>=0.13.2?
scikit-image>=0.9.3?
matplotlib>=1.3.1?
ipython>=3.0.0?
h5py>=2.2.0?
leveldb>=0.191?
networkx>=1.8.1?
nose>=1.3.0?
pandas>=0.12.0?
python-dateutil>=1.4,<2?
protobuf>=2.5.0?
python-gflags>=2.0?
pyyaml>=3.10?
Pillow>=2.3.0?
six>=1.1.0

sudo apt-get install python-numpy #之前以為和下面pip安裝的重復了，但是不安裝編譯時會報numpy的錯誤。 sudo pip install -r ${CAFFE_ROOT}/python/requirement.txt

• 1
• 2

2.編譯 pycaffe

cd ${CAFFE_ROOT} make clean make make pycaffe

• 1
• 2
• 3
• 4

3.繪制網絡結構圖

cd ${CAFFE_ROOT} python ./python/draw_net.py ./models/bvlc_reference_caffenet/train_val.prototxt ~/Desktop/caffenet.png

• 1
• 2
• 3

4.“dot” not found in path 的問題

2016.12.01添加：

先安裝graphviz否則會出現類似：“dot” not found in path 的問題?
安裝graphviz不要用pip install安裝，否則還是會找不到可執行程序?
安裝：sudo apt-get insall graphviz?
然后安裝pydot：pip install pydot?
其中pyparsing會自動安裝

摘自：pydot 安裝 用于caffe畫圖

轉載于:https://my.oschina.net/lilinzero/blog/884699

總結

以上是生活随笔為你收集整理的Caffe 网络结构可视化的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。