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就是绘制的图形