caffe绘制网络结构图

刚刚转战深度学习，配置好CPU下的caffe，测试mnist手写识别过后想进行一下网络结构图的可视化，有一个直观的认识，于是跳到实战caffe可视化方法一章进行学习

首先准备python环境

sudo apt-get update

sudo apt-get install python-pip python-dev python-numpy

sudo apt-get install gfortran

sudo pip install -r  {caffe root}/python/requirement.txt

sudo pip install pydot

之后编译pycaffe

cd {caffe root}

make clean

make

make pycaffe

正常按书上步骤只需到caffe的python目录下

 python draw_net.py ../models/bvlc_reference_caffenet/train_val.prototxt caffenet.png 

但接下来绘制网络结构图的时候却出现了错误

error提示是版本错误

于是进入python查看版本

python版本为1.8.2看错误提示应该要1.10或以上版本，于是重新安装numpy进行升级

这种方法却不好使

于是使用

成功升级到1.12.1

升级之后还是有错误，好事多磨啊

之后查找资料还需安装 sudo apt-get install graphviz 

最后终于画图成功，附上结果

这是caffenet的网络结构图

下面的是正在看的一篇深度学习边缘检测文章的网络结构图

刚刚接触caffe和python遇到错误还是卡了挺久的，写出来和大家分享，希望能有所帮助～