ubuntu14使用python绘制caffe的深度学习网络结构

来源:互联网 发布:作业盒子软件怎么样 编辑:程序博客网 时间:2024/05/06 09:42

1.配置python

     在安装caffe时未配置python,现在需要配置caffe文件中的Makefile.config文件,配置python的安装路径(默认即可),配置python依赖包:

sudo apt-get install python-pip

再转到caffe的python目录,安装scipy

cd caffe/pythonsudo apt-get install python-numpy python-scipy python-matplotlib ipython ipython-notebook python-pandas python-sympy python-nosesudo 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 sufor req in $(cat requirements.txt); do pip install $req; done
回到caffe目录,然后编译

<span style="font-size:18px;">make -jmake pycaffe</span>
检测是否成功配置python
<span style="font-size:18px;">pythonimport caffe</span>   //若无错误则安装成功,进行下面的步骤

2.安装pydot

python的pydot模块是Graphviz的dot语言的python接口,主要使用Graphviz绘图,

sudo apt-get updatesudo apt-get install python-pydot

3.安装Graphviz

sudo apt-get install graphvizsudo apt-get update

4.绘制net

     切换到caffe安装目录中的Python文件夹,检查是否存在draw_net.py文件,若存在(不存在说明前面出现安装过程错误)则执行命令

python draw_net.py ../examples/mnist/lenet_train_test.prototxt lenet.png  //lenet.png就是绘制的图形





0 0
原创粉丝点击