深度学习Caffe实战笔记(20)Windows平台 Faster-RCNN 训练自己的数据集

本文档详细介绍了如何在Windows环境下使用Faster R-CNN进行深度学习训练,重点在于配置数据集、修改标签、调整验证参数和网络模型参数,以及解决训练中遇到的问题。首先,将数据集放入指定位置并保持VOC2007命名,接着修改数据标签和验证参数,然后下载并调整模型文件和solver文件,最后执行训练脚本开始训练。
摘要由CSDN通过智能技术生成

昨天晚上博主干到12点多,终于用了一晚上时间搞定了Faster-Rcnn训练自己的数据集,这篇博客介绍如何用faster_rcnn训练自己的数据集,前提是已经准备好了自己的数据和配置好了faster-rcnn的环境。
制作数据集教程:http://blog.csdn.net/gybheroin/article/details/72581318
环境配置教程:http://blog.csdn.net/gybheroin/article/details/72428582
请大家自行前往查看。

1、把自己的数据集放到Faster-master中
我觉得这个过程是最重要的一个过程吧,博主在这里跳了很多的坑,最后找到了一个非常简单高效的方法。不用修改任何文件,就可以轻松实现加载自己的数据集。
在faster_rcnn-master文件夹下,有一个datasets文件夹,把VOCdevkit2007文件夹放在datasets文件夹下,然后在VOCdevkit2007文件夹下把你的数据集文件夹拷进去,并把你的数据集文件夹命名为VOC2007。通过这种方式可以最大限度的减少路径修改的麻烦。当然,如果你不想这样做,就需要修改faster_rcnn-master\datasets\VOCdevkit2007\VOCcode\VOCinit.m

这里写图片描述

不采用命名VOC2007的方法就要把这几个路径修改成自己的路径。

2、修改数据标签
在faster_rcnn-master\datasets\VOCdevkit2007\VOCcode\VOCinit.m中把标签修改成自己的标签

if VOC2006

    % VOC2006 classes

    VOCopts.classes={
  ...
        'bicycle'
        'bus'
        'car'
        'cat'
        'cow'
        'dog'
        'horse'
        'motorbike'
        'person'
        'sheep'};
else

    % VOC2007 classes

    VOCopts.classes={
  ...
         'person'      %修改成自己的标签
        };
end

3、修改验证参数
function\fast_rcnn\fast_rcnn_train.m和function\rpn\proposal_train.m中,

 ip.addParamValue('val_iters',       56,            @isscalar); %原来的值是500
    ip.addParamValue('val_interval',    2000,           @isscalar); 

据说这个地方是根据数据集来的,是验证集的1/5,博主没有试过不修改是什么情况,感兴趣的同学可以不修改对比有什么差别。

4、网络模型参数的修改

models\ fast_rcnn_prototxts\ZF\ train_val.prototxt models文件中默认的应该是空文件夹,这个模型大小1.4G,我没法上传,所以大家先去网上找找下载一下,如果找不到联系我我再把需要的文件发给你。

input: "bbox_targets"  
input_dim: 
评论 9
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值