三维计算机视觉 —— 中层次视觉 —— RCNN Family

  RCNN是从图像中检测物体位置的方法,严格来说不属于三维计算机视觉。可是这种方法却又很是很是重要,对三维物体的检测很是有启发,因此在这里作个总结。网络

一、RCNN - the original idea

  —— <Rich feature hierarchies for accurate object detection and semantic segmentation>ide

    这篇文章提出了用CNN网络来对物体进行检测的思路。函数

Q:优化

  a. CNN网络中存在卷积层和池化层,每次池化都会弱化物体的位置信息,强化物体的特征信息,因此CNN网络最终会告诉咱们是什么,而不是在哪儿idea

  b. 要使用CNN网络来检测,直觉上咱们能够训练一个识别某物体的网络,来对小方块进行分类。可是这须要大量的训练集,可能对于待检测物体,咱们没有收集大量训练集的机会spa

  c. CNN网络的图像输入层具备固定的维度,任意大小的小方块是没法直接输入到CNN网络里的设计

A:blog

  a. 文章提出了能够在已经训练好的网络上利用小规模的训练集进行优化,也能达到很好的效果。图片

  b.经过 selective search来肯定可能含有物体的小方块it

  c.将消方块进行拉伸(warp),送入CNN进行分类,最终实现检测

细节:

  一、使用了ILSVRC 2012对网络进行预训练,步长0.01

  二、精训练使用的步长是0.001

  三、mini-batch size = 128, 其中背景96,带东西的32。有意的bias,让网络更大几率判为背景

  四、loss function 中,IoU超过50%判为1,不然为0

  五、使用svm对物体类别进行判断

  六、对box进行线性回归,得到更高的精度(后面还会提到)

二、SPPNet - 结合金字塔

  —— <Spatial pyramid pooling in deep convolutional networks for visual recognition>

  在第一步使用rcnn中,检测须要将图片拉伸成特定的大小,便于输入网络。这显然很不合理,不少东西拉伸之后就彻底变形了,这会下降检测的精度。因此须要一种不拉伸方块的方法,来对物体进行检测。

Q:

  a.拉伸图像会带来识别精度的降低,在r-cnn中尤为明显

A:

  a.金字塔池化:将任意维度的图片池化成同一维度。例如,图片为256*256,金字塔接受的就是maxpool2dlayer(16,'stride',16),若是图片为128*128,金字塔接受的就是maxpool2dlayer(8,'stride',8)

  

细节:

  只用了两种bin size 来构建训练集,180 * 180, 224 * 224. 由于只有两种状况,因此能够很方便的构建bp函数

三、FastCNN - 要啥金字塔,一层就够啦

  —— <Fast R-CNN>

Q: 

  a.对于每一个 box/proposal 都须要进行一次feed forward.

  b.训练时须要对分类函数和回归函数分开训练

A:

  a.对图像进行一次总体的feed forward,获得总的卷积结果;对总结果中的box,每一个box一次,分别进行ROI pooling (其实就是金字塔的第一层)

  b. ROI pooling 后会获得固定维度的向量,送入屡次全链接层,直接映射成类别和box的回归

细节:

  一、训练时,全部的权重都获得迭代

 

四、Faster R-CNN — 这才是颠覆

  ——<Faster R-CNN: Towards Real-Time Object Detection with Region Proposal Networks>

Q:

  a. 对每张图都要进行一次 selective search 以得到proposal太耗时了

A:

  a.设计一个神经网络 Region Proposal Net 自动来提出proposal 吧!

RPN 本质上能够看做一个很是独立的部分,虽然它号称使用了整幅图像的feature,但功能上RPN是独立的。RPN的任务是输入图像,输出一系列Proposal(四点坐标+是否有物体)

一、使用卷积网络的最后一个卷积层卷积完的结果 Last_Image * final_convolution2dlayer(n,d). 做为RPN的真正输入(前面都是公用的,后面才是RPN)

二、使用一个3*3的小卷积掩模将1中的结果抽象成d 维向量。

三、将该向量送入2个全链接层——分类全链接(2K输出)和坐标全链接(4K输出)————k个proposal (3 scale,2 aspect ratio)

四、在2中,掩模的位置决定了全链接输出参数的"原点"

 

五、总结

  在神经网络中,全链接是最强的非线性映射方式,也是花费最重的。在结果层少许的使用全链接以换取强大的映射方程是颇有意义的。

  说了那么多,和三维视觉有毛关系?其实关系在这里,在二维图像中检测物体位置和在三维图像中检测物体位姿是对偶的。不信?见 <Deep Sliding Shapes for amodal 3D object detection in RGB-D images>

相关文章
相关标签/搜索