淘先锋技术网

首页 1 2 3 4 5 6 7

Caffe win10+VS2019环境配置及python接口配置

在win10+vs2019环境下成功编译运行了caffe-windows(cpu-only)

并配置好了编译生成的python接口的使用

开发环境

  • 系统:Win10
  • IDE:VS2019

Caffe环境配置及编译

参考:https://blog.csdn.net/waji2000/article/details/105155198

所需下载项:https://pan.baidu.com/s/19mmhrDkqKS4FjBmh7oPhQg 提取码:dv4f

caffe环境配置的难点就是有很多依赖库要装,原来如果找不到依赖库会自动下载,放到vs2019容易出错版本不匹配等等。这个博客的作者提供了bulider解决的这个问题。

这样过程就很简单了,只需要

  1. 配置编译好bulider,解决依赖库的问题,将原配置文件里自动下载的部分删掉,换为从bulider路径里获取。
  2. cmake配置生成vs2019解决方案(python路径我换成了自己anaconda下的python2.7)
  3. 打开vs2019编译即可

编译

vs基本的编译模式分为release和debug。

两者生成的文件名称不同,debug的文件名会多一个-d。

例如debug编译生成caffe-d.exe,release编译生成caffe.exe

生成的python接口名字也不同,debug下文件名为_caffe-d.pyd,release为_caffe.pyd

运行的时候要注意这点。

测试

到release目录下运行生成的exe看看(anaconda多版本python记得切换到对应的环境运行,比如我的默认环境为python3直接运行会报缺少python27.dll)

ps:我cmake生成解决方案的路径是caffe-windows\bulid
请添加图片描述

python接口配置

编译完成后的python接口存储在..\caffe-windows\python\路径下的caffe文件夹中。
在这里插入图片描述

将整个文件夹复制到所使用python环境根目录..\Lib\site-packages\下即可。

用python运行import caffe测试

可能会报某些库缺失,例如scipy、google protobuff等等,根据提示pip install依次安装好。

然后就可以在自己的python环境里使用caffe的python接口了。

cmd测试draw_net.py

> cd 你的路径/caffe-windows/python
> python draw_net.py ../models/bvlc_reference_caffenet/deploy.prototxt ../models/bvlc_reference_caffenet/structure_deploy.jpg

在这里插入图片描述

会提示缺失库需要conda install graphvizconda install pydotplus

生成的caffenet结构图如下

在这里插入图片描述