1.3. svm分类
通过上述卷积神经网络获取候选区域的特征向量,进一步使用svm进行物体分类,关键知识点如下:
使用了一个svm进行分类:向svm输入特征向量,输出类别得分
用于训练多个svm的数据集是imagenet数据
将2000×4096维特征(2000个候选框,每个候选框获得4096的特征向量)与20个svm组成的权值矩阵4096×20相乘(20种分类,svm是二分类器,每个种类训练一个svm,则有20个svm),获得2000×20维矩阵表示每个建议框是某个物体类别的得分
分别对上述2000×20维矩阵中每列即每一类进行非极大值抑制剔除重叠建议框,得到该列即该类中得分最高的一些候选框;
svm训练:
样本:
由于svm是二分类器,需要为每个类别训练单独的svm;
svm训练时,输入正负样本是在alexnet cnn网络输出的4096维特征向量,输出为该类的得分
由于负样本太多,采用hard negative mining的方法在负样本中选取有代表性的负样本
1.4 边框修正
使用一个回归器进行边框回归:输入为卷积神经网络pool5层的4096维特征向量,输出为x、y方向的缩放和平移,实现边框的修正。在进行测试前仍需回归器进行训练。 回归器训练
样本:
在2014年r-cnn横空出世的时候,颠覆了以往的目标检测方案,精度大大提升。对于r-cnn的贡献,可以主要分为两个方面: 1) 使用了卷积神经网络进行特征提取 2) 使用bounding box regression进行目标包围框的修正 但是,我们来看一下,r-cnn有什么问题: 1) 耗时的selective search,对一张图像,需要花费2s 2) 耗时的串行式cnn前向传播,对于每一个候选框,都需经过一个alexnet提取特征,为所有的候选框提取特征大约花费47s 3) 三个模块(cnn特征提取、svm分类和边框修正)是分别训练的,并且在训练的时候,对于存储空间的消耗很大
2. fast r-cnn
面对r-cnn的缺陷,ross在2015年提出的fast r-cnn进行了改进,下面我们来概述一下fast r-cnn的凯发天生赢家一触即发官网的解决方案:
首先还是采用selective search提取2000个候选框roi
使用一个卷积神经网络对全图进行特征提取
使用一个roi pooling layer在全图特征上摘取每一个roi对应的特征
分别经过为21和84维的全连接层(并列的,前者是分类输出,后者是回归输出) fast r-cnn通过cnn直接获取整张图像的特征图,再使用roi pooling layer在特征图上获取对应每个候选框的特征,避免了r-cnn中的对每个候选框串行进行卷积(耗时较长)。
2.1 roi pooling layer
对于每个roi而言,需要从共享卷积层获取的特征图上提取对应的特征,并且送入全连接层进行分类。因此,roi pooling主要做了两件事,第一件是为每个roi选取对应的特征,第二件事是为了满足全连接层的输入需求,将每个roi对应的特征的维度转化成某个定值。roi pooling示意图如下所示:
如上图所示,对于每一个roi,roi pooling layer将其映射到特征图对应位置,获取对应特征。另外,由于每一个roi的尺度各不相同,所以提取出来的特征向量region proposal维度也不尽相同,因此需要某种特殊的技术来做保证输入后续全连接层的特征向量维度相同。roi pooling的提出便是为了解决这一问题的。其思路如下:
将region proposal划分为目标h×w大小的分块
对每一个分块中做maxpooling(每个分块中含有多个网格,每个分块获取一个特征值)
将所有输出值组合起来便形成固定大小为h×w的feature map
fast r-cnn的贡献可以主要分为两个方面:
取代r-cnn的串行特征提取方式,直接采用一个cnn对全图提取特征(这也是为什么需要roi pooling的原因)。
除了selective search,其他部分都可以合在一起训练。 fast r-cnn也有缺点,体现在耗时的selective search还是依旧存在。
下一篇:父母“婴语”好 孩子更健谈