Caffe-SSD(single-shot-multibox)在Windows平台下的配置(CPU和GPU)与训练记录

本文详细记录了在Windows 10上配置Caffe-SSD(CPU和GPU)的过程,包括编译Caffe-SSD、解决GPU编译问题以及训练SSD模型的步骤。配置中涉及CUDA 8.0、TensorFlow 1.13.1、Python 2.7.13等,训练过程中解决了训练损失函数异常等问题,最终实现了SSD模型的训练和检测。
摘要由CSDN通过智能技术生成

1在windows上面训练和测试ssd的方法很少,而且配置也相当之麻烦,搞了几天才调试成功。现在写一下过程分享到博客上,顺便也可以作为今后再次安装的参考。我的配置环境:

windows: 10
gpu: Titan X
cuda: 8.0
caffe: caffe-ssd-microsoft
python: 2.7.13

Caffe-SSD的编译与配置

编译这部分不打算详细展开,网上有很多教程,CPU的较多,下面会给出好用的参考教程。这里推荐一下caffe-ssd-microsoft版本而不是caffe-windows加上caffe-ssd的修改版(网上一些教程是这么改的),两个方法都可以,我这里采用的就是单用caffe-ssd-microsoft的版本。

caffe-ssd-microsoft下载:

https://github.com/conner99/caffe

配置的参考博客链接:

http://blog.csdn.net/buaalei/article/details/54668507
http://www.jianshu.com/p/9a84cc434e05

这两个可以配合着看,尤其GPU部分推荐看第二个博客。反正我前前后后不停地编译不停地改。

配置caffe-ssd-microsoft

编译libcaffe模块

这个就参考上面给的博客就可以了,我就遇到下面几个问题:
1. 添加3rdparty的hungarian.h和cpp
在这里找这个两个文件,链接:http://pan.baidu.com/s/1mhYuf7y 密码:3jp2
2. 注释掉 detection_output_layer.hpp和detection_output_layer.cu和detection_output_layer.cpp有关regex 的引用和语句
3. C/C++ –> 常规 里的将警告提示为错误修改 为

编译caffe模块

libcaffe编译成功后,再进行caffe的编译,注意点就是要添加opencv的引用目录和库目录。这个不难。

编译pycaffe模块

这个可选,如果想用python接口的话这个是必须编译的,注意的是首先要配置好下面两个库的引用和库目录
1. python库目录
2. opencv库目录
之后进行google的protobuf的配置,这个不配置会报错的。

配置protobuf

配置参考:

http://www.jianshu.com/p/0c563b2c0fdb

  1. 选择python2的环境
  2. 下载protobuf.3.0.0:
    下载链接:

    https://github.com/google/protobuf/releases/tag/v3.0.0

    下载两个包:protobuf-python-3.0.0.zip 以及 protoc-3.0.0-win32.zip
    protobuf-python-3.0.0为protobuf的安装包
    protoc-3.0.0-win32包含protobuf的编译器protoc的win32版本,用以编译*.proto文件。

下载后将protoc-3.0.0-win32\bin\protoc.exe 复制到protobuf-3.0.0\src\
用cmd进入到protoc-3.0.0\python目录,运行 python setup.py build,生成大量*.py,如下命令

    python setup.py build

然后再分别写入下面两个命令就可以了。

  • 12
    点赞
  • 33
    收藏
    觉得还不错? 一键收藏
  • 42
    评论
评论 42
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值