关闭

faster rcnn在自己的数据集上训练

141人阅读 评论(0) 收藏 举报
分类:

本文是一个总结,参考了网上的众多资料,汇集而成,以供自己后续参考。

一般说来,训练自己的数据,有两种方法:第一种就是将自己的数据集完全改造成VOC2007的形式,然后放到py-faster-rcnn/data 目录下,然后相应地改变相应模型的参数,比如种类等。 data目录下存放的数据如下:

VOCdevkit2007
	└── VOC2007
		├── Annotations
		│   └── *.xml
		│
		├── ImageSets
		│   └── Main
		│       ├── train.txt
		│       └── test.txt
		└── JPEGImages
    			└── *.jpg
这样做最省事。


第二种,基本上也是改造成VOC2007的形式,然后放到py-faster-rcnn/data 目录下,但是允许Annotations不按照xml的形式。可以是自己的形式

YOUR_DATASET_devkit
	└── data
		├── Annotations
		│   └── *.txt
		│
		├── ImageSets
		│       ├── train.txt
		│       └── test.txt
		└── JPEGImages
    			└── *.jpg or *.png

关于第二种,可以参考如下文章:

1. deboc/py-faster-rcnn  这个代码是py-faster rcnn的拷贝,实现了二分类模型。里面的data组织如下:

INRIA_Person_devkit/
|-- data/
    |-- Annotations/
         |-- *.txt (Annotation files)
    |-- Images/
         |-- *.png (Image files)
    |-- ImageSets/
         |-- train.txt
按照教程:https://github.com/deboc/py-faster-rcnn/blob/master/help/Readme.md 可以训练行人检测。

另外《Fast RCNN 训练自己数据集 (2修改数据读取接口)》 也实现了自己的数据读取。


下面重点说一下第一种。

具体的细节可以参考: http://blog.csdn.net/Gavin__Zhou/article/details/52052915

需要注意的是 根据文章<解决faster-rcnn中训练时assert(boxes[:,2]>=boxes[:,0]).all()的问题>   我们对lib/datasets/imdb.py,append_flipped_images()函数修改如下:

 def append_flipped_images(self):
        num_images = self.num_images
        widths = self._get_widths()
        for i in xrange(num_images):
            boxes = self.roidb[i]['boxes'].copy()
            oldx1 = boxes[:, 0].copy()
            oldx2 = boxes[:, 2].copy()
            boxes[:, 0] = widths[i] - oldx2 - 1
            boxes[:, 2] = widths[i] - oldx1 - 1
            for b in range(len(boxes)):
                if boxes[b][2]< boxes[b][0]:
                    boxes[b][0] = 0
            assert (boxes[:, 2] >= boxes[:, 0]).all()
            entry = {'boxes' : boxes,
                     'gt_overlaps' : self.roidb[i]['gt_overlaps'],
                     'gt_classes' : self.roidb[i]['gt_classes'],
                     'flipped' : True}
            self.roidb.append(entry)
        self._image_index = self._image_index * 2


我们说一说其他的,

其他参考文档:

1. 使用Faster-Rcnn进行目标检测(实践篇)

2. Faster rcnn 安装、训练、测试、调试

3. matlab训练样本集的制作

4.  制作自己的数据集用于faster-rcnn训练

5. 将数据集做成VOC2007格式用于Faster-RCNN训练   (很重要)

6. Faster-RCNN+ZF用自己的数据集训练模型(Python版本)

7. Training'R CNNs'of'various'velocities    训练faster rcnn的官方文档1

8. 训练faster rcnn的官方文档2

9. Faster-RCNN+ZF用自己的数据集训练模型(Matlab版本)

10. Py-faster-rcnn实现自己的数据train和demo  (很重要)


















0
0
查看评论
发表评论
* 以上用户言论只代表其个人观点,不代表CSDN网站的观点或立场

Caffe学习系列——Faster-RCNN训练自己的数据集

2016-10-23 Deep Learning►Caffe Caffe学习系列——6使用Faster-RCNN进行目标检测 Contents 1. 配置与运行De...
  • CV_adventurer
  • CV_adventurer
  • 2017-05-29 23:38
  • 4103

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

昨天晚上博主干到12点多,终于用了一晚上时间搞定了Faster-Rcnn训练自己的数据集,这篇博客介绍如何用faster_rcnn训练自己的数据集,前提是已经准备好了自己的数据和配置好了faster-...
  • gybheroin
  • gybheroin
  • 2017-05-21 09:53
  • 1959

Faster RCNN 训练自己的数据集

Faster RCNN 训练自己的数据集
  • Apple__fly
  • Apple__fly
  • 2016-06-19 14:55
  • 2561

Faster-RCNN+ZF制作自己的数据集和训练模型完整文件及教程(Matlab版本)

  • 2017-07-09 11:06
  • 8.63MB
  • 下载

基于Faster-RCNN-TF的gpu运行总结(自己准备数据集)(训练篇)

最近经过一段对tensorflow和faster-rcnn的学习,并且亲身去跑了两个不同框架下的faster-rcnn代码,所以就在这里做一下总结。 这里,我就主要记录一下自己在跑tenso...
  • qq_39531954
  • qq_39531954
  • 2017-12-21 17:01
  • 120

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

昨天晚上博主干到12点多,终于用了一晚上时间搞定了Faster-Rcnn训练自己的数据集,这篇博客介绍如何用faster_rcnn训练自己的数据集,前提是已经准备好了自己的数据和配置好了faster-...
  • gybheroin
  • gybheroin
  • 2017-05-21 09:53
  • 1959

使用faster rcnn训练imageNet上的部分数据集

最近在做毕业设计,需要用到faster rcnn,训练数据使用的是我自己下载的imageNet里的部分数据,主要用到了其中的bike,motorbike,car,bus,train等交通工具。由于im...
  • u014696921
  • u014696921
  • 2016-10-27 21:35
  • 3904

Faster-RCNN/SSD/训练将数据集做成VOC2007格式

reference: http://blog.csdn.net/sinat_30071459/article/details/50723212 0.文件夹名 首先,确定你的数据集所放...
  • u011534057
  • u011534057
  • 2016-06-30 14:50
  • 5405

Faster-RCNN+ZF用自己的数据集训练模型(Python版本)

一,做自己的数据集 0.文件夹名 首先,确定你的数据集所放的文件夹名字,例如我的叫153(或者和voc2007一样的名字:VOC2007) 1.图片命名 虽然说图片名对训练没什么影响,但建...
  • zhangtong11111
  • zhangtong11111
  • 2017-12-19 14:30
  • 73

Faster-RCNN+ZF用自己的数据集训练模型(Python版本)

说明:本博文假设你已经做好了自己的数据集,该数据集格式和VOC2007相同。下面是训练前的一些修改。 (做数据集的过程可以看http://blog.csdn.net/sinat_30071459/a...
  • qq_36673141
  • qq_36673141
  • 2017-12-27 09:04
  • 32
    个人资料
    • 访问:285087次
    • 积分:4126
    • 等级:
    • 排名:第8660名
    • 原创:117篇
    • 转载:82篇
    • 译文:8篇
    • 评论:106条
    个人网站
    最新评论