在海康实习的第一周,给我分配了任务,因为要放到AI芯片上跑,需要用到caffe,于是就开始了我的caffe学习之路,从caffe安装上我就清楚的感受到这东西不是善茬,相较于tensorflow、pytorch要麻烦的多。
安装过程参考:https://blog.csdn.net/weixin_37621229/article/details/80547934
由于项目还不需要使用到GPU,因此只安装了CPU版本
1、下载caffe 源码: https://github.com/Microsoft/caffe,保存文件夹为 caffe_master
2、打开caffe_master目录下的windows文件夹,将CommonSettings.props.example复制一份为CommonSettings.prop,这个文件是修改配置用的,将CpuOnlyBuild设置为true,将python support设置为true,并更改第48行的python路径,这个路径是你自己的python路径,最好安装一个ananconda,因为需要用到各种包,anaconda一步搞定,需要注意的是caffe最好使用py2.7,你可以新建一个python2.7的环境。
3、双击Caffe.sln文件,这是caffe的工程,首先点击libcaffe,右击解决方案/配置属性/配置:Release,平台:活动(X64),打开libcaffe属性设置,关闭视警告为错误(即设置为No)如果不设置的话,在编译boost库的时候会由于文字编码的警告而报错!!!,这样生成后发现caffe-master目录下自动多了一个文件夹NugetPackages!,Nuget工具可以帮助你自动安装很多caffe的依赖包。
4、回到caffe工程,对你需要的部分重复上述过程,如你需要python接口,就可以在pycaffe上重复上述过程,会发现出现了一个Build文件夹。
5、配置python接口:上述过程搞定之后,我们进入Build/x64/Release/pycaffe,将caffe文件夹复制到你的python2.7文件夹下的lib/site-package,之后在命令行中进入你的python2.7环境,import caffe,如若报错:ImportError:No module named google.protobuf.internal,那么pip install protobuf即可解决!!!