ubuntu14使用python繪製caffe的深度學習網絡結構

1.配置python

     在安裝caffe時未配置python,現在需要配置caffe文件中的Makefile.config文件,配置python的安裝路徑(默認即可),配置python依賴包:

sudo apt-get install python-pip

再轉到caffe的python目錄,安裝scipy

cd caffe/python
sudo apt-get install python-numpy python-scipy python-matplotlib ipython ipython-notebook python-pandas python-sympy python-nose
sudo apt-get install python-numpy python-scipypython-matplotlib python-sklearn python-skimage python-h5py python-protobufpython-leveldb python-networkx python-nose python-pandas python-gflags Cythonipython 

最後安裝requirement裏面的包,需要root權限

sudo su
for req in $(cat requirements.txt); do pip install $req; done
回到caffe目錄,然後編譯

<span style="font-size:18px;">make -j
make pycaffe</span>
檢測是否成功配置python
<span style="font-size:18px;">python
import caffe</span>   //若無錯誤則安裝成功,進行下面的步驟

2.安裝pydot

python的pydot模塊是Graphviz的dot語言的python接口,主要使用Graphviz繪圖,

sudo apt-get update
sudo apt-get install python-pydot

3.安裝Graphviz

sudo apt-get install graphviz
sudo apt-get update


4.繪製net

     切換到caffe安裝目錄中的Python文件夾,檢查是否存在draw_net.py文件,若存在(不存在說明前面出現安裝過程錯誤)則執行命令

python draw_net.py ../examples/mnist/lenet_train_test.prototxt lenet.png  //lenet.png就是繪製的圖形





發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章