deep learning 目标检测(三)之Fast R-CNN

Abstract

特点:有一个多任务的loss,简化了学习过程,并且提高了效率。比传统的R-CNN和SPP-Net更加快,而且精度高。

1. Introduction

对目标的精确定位目前还存在两大挑战:

       第一:大量的候选区域需要被估计。为了精度定位,卷积层的特征需要共享之前他在原图像中的位置。然后这个方法限制了后向传播的误差,接着潜在的影响了精度。

       第二:候选对象位置需要是精练来达到高精度。传统的R-CNN和SPP-net以前的这个挑选候选区域过程在一个独立于的学习过程的。

本篇文章:提出了一个单一过程的学习过程。在学习过程中,同时找出那些候选区域并且能精炼他的空间位置。并且本文还提出一个新的方法:在学习过程来共享卷积特征,允许将全部的反向误差都传播回来,来增加精度。除此之外还用了截断SVD等等一些手段来增加精度并且加快速度。

1.1. RCNN and SPPnet

具体看前两篇。

1.2. Contributions


advantages

                                                                     code

2. Fast R-CNN training


model 结构

快速R-CNN网络将整个图像一组object proposals作为输入。

       网络首先使用几个卷积(conv)和最大池层来处理整个图像,以产生conv feature map。

       然后,对于每个对象proposal,感兴趣区域(RoI)池层从特征图中抽取固定长度的特征向量。

       每个特征向量被馈送到完全连接(fc)层序列,其最终分支成两个同级输出层:

               一个产生对K个对象类加上全部捕获的“背景”类的softmax概率估计(one that produces           softmax probability estimates over K object classes plus a catch-all “background” class)

                 另一个对每个K对象类输出四个实数,每组4个值编码提炼定义K个类中的一个的的              边 界框位置。(another layer that outputs four real-valued numbers for each of the K                  object classes. Each set of 4 values encodes refined bounding-box positions for one of            the K classes.)

2.1. The RoI pooling layer

The region of interest (RoI) pooling layer是简化版的spatial pyramid pooling(SPP-net).因为(RoI) pooling layer只有单一的尺度。

Rol pooling layer的作用主要有两个:

一个是将image中的RoI定位到feature map中对应patch

另一个是用一个单层的SPP layer将这个feature map patch下采样为大小固定的feature再传入全连接层。

RoI池层使用最大池化将任何有效的RoI区域内的特征转换成具有H×W(例如,7×7)的固定空间范围的小feature map,其中H和W是层超参数它们独立于任何特定的RoI。

在本文中,RoI是conv feature map中的一个矩形窗口。

每个RoI由定义其左上角(r,c)及其高度和宽度(h,w)的四元组(r,c,h,w)定义。

RoI层仅仅是Sppnets中的spatial pyramid pooling layer的特殊形式,其中只有一个金字塔层.

2.2. Using pretrained networks

之前始于训练截断我们训练了3个Image network.当提个预训练模型来初始化FRCN的时候,需要进行3个转变。

结构变了

1.最后一层最大池化层变成了RoI池化层,可以为全连接层产生固定大小的feature。

2.两个sibling层,同时实现不同的功能。

输入变了

3.a batch of N images 和a list of R RoIs.batch的大小和RoI的输出量是动态变化的

2.3. Finetuning for detection

limitation:在3层的softmax classifier 在检测的时候进行了微调。因为3层的softmax classifier接受固定长度的feature向量。它是来自spatial pyramid pooling。在此之前的网络结构卷积层工作方式是offline的。所以在微调阶段,反向传播不可能到达卷积层。换句话说在微调阶段:只有3层的classifier被更新了参数。

由于以上限制limitation的存在:在SPPnet中SGD采用的方式是,在众多的RoI候选区域(将所有图片的RoI区域放在一起)进行采样。所以每一个Minibatch里面的RoI可能是来自不同图片的。

为了消除这个限制limitation:本文采取的SGD策略是有等级的采样:先对输入图片进行采样,然后在已经采样得到的图片中采样得到RoI.这些来自同一个通篇的ROI对于计算和储存是很有效果的。

Multi-task loss.


LOSS合并

Mini-batch sampling.

Back-propagation through RoI pooling layers.


SGD hyper-parameters.

解析博客

2.4. Scale invariance

SPPnet用了两种实现尺度不变的方法:

1. brute force (single scale),直接将image设置为某种scale,直接输入网络训练,期望网络自己适应这个scale。

2. image pyramids (multi scale),生成一个图像金字塔,在multi-scale训练时,对于要用的RoI,在金字塔上找到一个最接近227x227的尺寸,然后用这个尺寸训练网络。

虽然看起来2比较好,但是非常耗时,而且性能提高也不对,大约只有%1,所以这篇论文在实现中还是用了1。

3. Fast R-CNN detection

输入分两种形式:

1.single-scale:单一尺度的图片和它的大约2000多张的候选图。

2.multi-scale:一张图片的金字塔和对应金字塔的候选区域。


3.1. Truncated SVD for faster detection

在分类中,计算全连接层比卷积层快,而在检测中由于一个图中要提取2000个RoI,所以大部分时间都用在计算全连接层了。文中采用奇异值分解的方法来减少计算fc层的时间。

SVD

矩阵知识

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 214,504评论 6 496
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 91,434评论 3 389
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 160,089评论 0 349
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 57,378评论 1 288
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 66,472评论 6 386
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 50,506评论 1 292
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 39,519评论 3 413
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 38,292评论 0 270
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 44,738评论 1 307
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 37,022评论 2 329
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 39,194评论 1 342
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 34,873评论 5 338
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 40,536评论 3 322
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 31,162评论 0 21
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,413评论 1 268
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 47,075评论 2 365
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 44,080评论 2 352

推荐阅读更多精彩内容