使用Pycaffe绘制Caffe网络图

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/tianrolin/article/details/52723300

检验Caffe能否在Python中使用,我们可以在Python中试着import caffe。

import caffe

如果成功的话就可以直接跳到文末看画图就好了,失败的话往往会显示:

ImportError: No module named _caffe

则说明caffe路径未设置正确,那么就一起来配置一下吧。我们需要先设定系统环境变量PYTHONPATH,是你的caffe目录下的python文件夹 $(caffe)\python。再将$(caffe)\Build\x64\Release\pycaffe\caffe 目录生成的所有文件拷贝到$(caffe)\python\caffe目录中。

我们需要下载安装google.protobuf,地址在google protobuf releases。下载其中的protobuf-python-3.1.0.zip和protoc-3.1.0-win32.zip,各自解压缩,并将win32包中的protoc.exe放在python包的setup.py同一目录下,再执行build命令。

python setup.py build

执行test,确认编译正确

python setup.py test

输出:

----------------------------------------------------------------------
Ran 590 tests in 1.079s

OK (skipped=7)

继续执行install(这步不可省略)

python setup.py install

输出:

Finished processing dependenceies for protobuf==3.0.0

设置成功后,我们在python中再试着import caffe,
skimage.io
提示缺少skimage.io模块,说明还需要添加scikit-image。

pip install scikit-image

结果又出错了,
scikit-image_error

这是什么鬼?Visual C++ 9.0 is required??怎么还需要VS2008?机器上装的VS2013都不行?好吧,还需要安装一个Micorsoft Visual C++ Compiler for Python 2.7的包,去微软官网下载一个吧
Microsoft Visual C++ Compiler for Python 2.7

结果你会发现,还需要安装lapack/blas等等各种包,简直是要抓狂。这里给大家推荐一个Python扩展库站点,
Unofficial Windows Binaries for Python Extension Packages
建议下载对应的whl,先安装Numpy + mkl,然后再分别安装Scipy、scikit-image、pydot、Graphviz。Graphviz需要安装一个msi包,下载地址为 Graph Visualization Software。其中如果提示找不到dot.exe,那么还需要将Graphviz2.38\bin添加到系统环境变量PATH中。这一切都安装完成后再去

import caffe

终于成功了!忙了大半天是不是都忘了我们今天是要做什么了。。。

不是说好要画图的嘛。。。
执行绘制网络图命令:

draw_net.py ..\examples\mnist\lenet_train_test.prototxt lenet.png

Lenet

猜你喜欢

转载自blog.csdn.net/tianrolin/article/details/52723300