Windows+VS2013+GPU caffe 网络结构可视化亲自安装
想非常直观地查看到caffe网络结构,那就得看看这篇配置文章,本以为配置挺简单的,但陆陆续续弄了好久才弄好,其实跟自己的心态还是有关系的,因为出现错误的时候就很烦躁,也不认真看网上的文章,来讲讲步骤吧~
参考文章:
caffe 使能python接口使用draw_net.py绘制网络结构图过程
http://blog.csdn.net/lemianli/article/details/53034432
1、下载anaconda并安装,别去官网下,太慢了,下载清华的镜像文件挺快的;安装完毕后,利用命令conda list去cmd测试安装是否成功。
https://mirrors.tuna.tsinghua.edu.cn/help/anaconda/
2、编译pycaffe项目,编译之前,修改一下配置文件;
3、修改环境变量,将 你的目录\caffe-master\Build\x64\Release\添加到path变量中去;
4、打开CMD
运行以下几条命令:
pip install protobuf
pip install pydot
pip install GraphViz
pip install scikit-image
pip install scipy
pip install numpy
5、任意选一个网络进行可视化,如我选择\caffe-master\models\bvlc_reference_caffenet里的deploy.prototxt复制到\caffe-master\python下,并在\caffe-master\python目录写一个bat脚本,内容如下:
python draw_net.py deploy.prototxt 1.png
6、双击上面的脚本,出现importError:no module named _caffe错误,解决办法是:
把\caffe-master\Build\x64\Release\pycaffe下的caffe文件夹替代\caffe-master\python下的caffe
7、再次双击上面的脚本,出现无法找到dot.exe错误,可能是上面的GraphViz安装或版本有误
先去这个网址http://www.graphviz.org/Download_windows.PHP将graphviz-2.38.msi下下来,进行安装,可以发现它的bin目录下有这个我们需要的dot.exe,将这个bin目录添加到系统的环境变量中去即可,再次在CMD中敲打命令pip install ygraphviz即可。完成后记得关机重启噢
8、又一次双击运行上面的脚本,成功了,真好~