我们在学习深度的网络的时候,为了能有更好的感性认识,可以使用draw_net.py将caffe net 结构绘制出来
安装python所需要的依赖
sudo apt-get install python-numpy python-scipy python-matplotlib ipython ipython-notebook python-pandas python-sympy python-nose
然后
sudo apt-get install graphviz #graphviz是一种很好用的画图工具
sudo su
pip install pydot #如果未安装pip,可以根据提示,自行安装
在安装完成后,我们发现还是不能绘图,那是因为我们使用了anaconda,所有我们需要将/usr/local/lib/python2.7/dist-packages 文件下的内容,全部复制到/home/XXX/anaconda2/lib/python2.7/site-packages(xxx为用户名) 文件下。
测试绘制mnist的LeNet网络层结构命令:
cd caffe/python
python draw_net.py ../examples/mnist/lenet_train_test.prototxt lenet.png --rankdir=TB
效果如下:
常遇问题
此时如果使用
sudo pip install pydot
会提示没有pip这个命令,故希望读者能按博客里的建议安装步骤来,先切换到root用户,然后执行安装命令,如果提示没有pip,就按照提示安装pip就可以了。