R-CNN(目标检测算法)介绍

最新推荐文章于 2024-02-12 16:08:15 发布

Sea__c

最新推荐文章于 2024-02-12 16:08:15 发布

阅读量2.1k

点赞数 15

文章标签： cnn 目标检测算法

本文链接：https://blog.csdn.net/Sea__c/article/details/133975605

版权

一、什么是目标检测

1.1 目标检测概念

目标检测 (Object Detection) 是计算机视觉领域中的一项任务，旨在从图像或视频中准确地定位和识别多个感兴趣的目标物体。目标检测不仅需要确定目标的类别，还要找到目标在图像中的位置.由于各类目标不同的外观，颜色，大小以及在成像时光照，遮挡等具有挑战性的问题，目标检测一直处于不断的优化和研究中.

1.2 目标检测步骤:

数据收集和标注：首先，需要收集包含目标对象的图像或视频数据集。这些数据集应该覆盖所需检测的目标类别，并且需要进行手动标注，即为每个目标在图像中标记边界框及对应的类别标签。
候选区域生成：通过使用候选区域生成算法（如选择性搜索、边缘框架、基于深度学习的区域建议网络等），从输入图像中生成一组可能包含目标的候选区域。这些候选区域是用来进一步分析和检测目标的区域。
特征提取：对于每个候选区域，使用卷积神经网络（CNN）或其他特征提取方法来提取区域的特征表示。这些特征通常是在预训练的网络上提取得到的，可以捕捉到图像中的局部和全局信息。
目标分类与回归：基于提取的特征，使用分类器（如支持向量机(SVM)、softmax分类器）进行目标分类，确定候选区域中的目标类别。同时，还进行边界框回归，以更准确地定位目标在候选区域中的位置和尺寸。
非极大值抑制：由于一个目标可能会在多个候选区域中被检测到，需要进行非极大值抑制。该过程通过选择具有最高置信度的目标检测结果，并消除重叠的检测结果来排除冗余。
后处理和可视化：最后，对于每个被保留的目标检测结果，可以进行后处理操作，如去除低置信度的检测结果、进一步分析目标属性等。同时，还可以将检测结果以边界框或标签的形式可视化，使其易于理解和解释。

边界框是什么?

边界框（bounding box）是在目标检测和物体识别中经常使用的一种表示方法，用于定位目标在图像中的位置和范围。边界框是一个矩形框，通常由左上角和右下角两个点的坐标表示。

边界框提供了目标在图像中的大致位置信息，并且可以通过其坐标来确定目标的尺寸和形状。在目标检测任务中，边界框通常与目标类别关联，用于表示检测到的目标及其对应的类别。

边界框可以简洁地描述目标的位置和区域，使得我们可以准确地标记、跟踪和定位目标。它是许多目标检测算法的核心组成部分，例如RCNN系列、YOLO、SSD等算法。

边界框通常用一组数值来表示，其中最常见的是四个浮点数或整数，分别表示左上角和右下角点的(x, y)坐标。有时候还会包括目标的类别标签和置信度得分等附加信息。

通过边界框，我们可以方便地进行目标的定位、裁剪、划分和测量等操作，为目标检测和物体识别任务提供了重要的信息基础。

1.3 目标检测算法分类

1.传统方法与深度学习方法:

传统方法：传统的目标检测方法主要基于手工设计的特征和机器学习算法，如Haar特征、HOG（Histogram of Oriented Gradients）和SIFT（Scale-Invariant Feature Transform）等。这些方法通常在目标检测性能方面存在一定的限制。
深度学习方法：近年来，随着深度学习的发展，基于深度神经网络的目标检测方法取得了显著的进展。这些方法通过端到端的训练，自动学习图像特征表示和目标分类器，例如RCNN系列、YOLO系列、SSD、RetinaNet和EfficientDet等。

2.两阶段方法与单阶段方法:

两阶段方法(Tow Stage)：两阶段方法将目标检测任务分为候选区域生成和目标分类定位两个阶段。首先，通过选择性搜索、区域建议网络（RPN）等方法生成候选区域，然后对每个候选区域进行分类和边界框回归。典型的两阶段方法有RCNN系列和Faster R-CNN等。
单阶段方法(One Stage)：单阶段方法直接在图像上密集地预测目标的类别和边界框，通常通过将图像分割为网格单元，并为每个单元生成预测来实现。常见的单阶段方法包括YOLO系列和SSD等。

3.基于特征金字塔(Feature Pyramid)的方法:

特征金字塔是指在不同尺度上提取特征并融合它们以检测不同大小的目标。基于特征金字塔的方法可以更好地处理多尺度目标。例如，FPN（Feature Pyramid Network）和RetinaNet就是基于特征金字塔的目标检测算法。

4.单目标检测与多目标检测：

单目标检测：单目标检测算法旨在检测和定位图像中的单个目标实例。
多目标检测：多目标检测算法能够同时检测和定位图像中的多个目标实例，例如目标检测中的行人、车辆、动物等.

二、什么是R-CNN

2.1 R-CNN概念

RCNN（Region-based Convolutional Neural Networks）是一种经典的目标检测算法，最初由Girshick等人在2014年提出。它通过将图像分割成多个候选区域，并对每个候选区域进行分类和边界框回归来实现目标检测。

2.2 R-CNN结构

整体步骤:

候选区域生成：在图像中提取候选区域，这些候选区域通常是通过选择性搜索（Selective Search）等算法生成的。选择性搜索将图像分割为多个不同尺度和形状的区域，并根据颜色、纹理、大小等特征进行合并，以产生可能包含目标的候选区域。
特征提取：对每个候选区域应用卷积神经网络（CNN）来提取特征表示。通常使用预训练的CNN模型，如AlexNet、VGGNet或ResNet等，在候选区域上进行前向传播，以获得固定长度的特征向量。
目标分类:提取的特征向量作为输入，经过一个支持向量机（Support Vector Machine，SVM）分类器进行目标类别的分类。在训练阶段，SVM使用正负样本对模型进行训练，学习区分目标和非目标的能力。
非极大值抑制：由于选择性搜索算法可能会生成重叠的候选区域，为了消除冗余的检测结果，R-CNN使用非极大值抑制（Non-Maximum Suppression，NMS）来筛选最终的检测边界框。NMS通过比较检测框之间的重叠程度和得分，并保留最具有代表性的框。
边框位置修正 Boundingbox Regression

2.2.1 候选区域生成-----选择性搜索（Selective Search)算法:

选择性搜索（Selective Search）是一种用于生成候选区域的图像分割算法，常用于目标检测任务中。它通过分割和合并策略生成多个大小、形状和纹理不同的候选区域，以涵盖可能包含目标对象的各种可能性。

以下是Selective Search算法的详细介绍：

分割策略：选择性搜索从原始图像开始，首先将图像分割成许多小区域。这些小区域可以基于像素颜色、纹理、亮度等特征进行分割。具体常用的分割方法有基于图像像素相似性的连通区域分割或均值漂移分割等。
合并策略：在分割步骤之后，选择性搜索使用合并策略来逐渐合并相似的区域。这一步骤能够根据相似性度量（如颜色直方图、纹理特征等）来衡量两个区域之间的相似程度，并将相似性最高的区域合并为一个更大的区域。这个过程迭代进行，直到只剩下一个整体区域。
候选区域生成：在合并完成后，选择性搜索得到了多个不同尺寸、形状和纹理的候选区域。这些候选区域可以覆盖可能包含目标对象的各种可能性，从而提供了多样化的选择。

选择性搜索算法的优点在于它能够生成多样性且具有不同尺度的候选区域，从而能够更全面地涵盖目标对象的各种变化。它能够适应不同图像场景和目标特征的变化，并且对于较复杂的图像具有较好的效果。

值得注意的是，选择性搜索是一种计算密集型算法，因为它需要对图像进行大量的分割和合并操作。为了提高效率，可以采用一些加速策略，如针对不同尺度的图像金字塔、减小分割区域数量等。

总结来说，选择性搜索是一种常用的目标检测中的区域提取算法，通过分割和合并策略生成多样性的候选区域。这些候选区域能够涵盖可能包含目标对象的各种变化，为后续的目标分类和定位提供了多样性的选择。

2.2.2 特征提取

1.网络架构选择

两个可选方案：第一选择经典的Alexnet；第二选择VGG16。经过测试Alexnet精度为58.5%，VGG16精度为66%。VGG这个模型的特点是选择比较小的卷积核、选择较小的跨步，这个网络的精度高，不过计算量是Alexnet的7倍。后面为了简单起见，我们就直接选Alexnet.

Alexnet特征提取部分包含了5个卷积层、2个全连接层，在Alexnet中pool5层神经元个数为9216、 fc6、fc7的神经元个数都是4096，通过这个网络训练完毕后，最后提取特征每个输入候选框图片都能得到一个4096维的特征向量.

2.有监督预训练阶段(训练网络参数)

在实际测试的时候，模型需要通过CNN提取出中的特征，用于后面的分类与回归。所以，如何训练好CNN成为重中之重。

由于物体标签训练数据少，如果要直接采用随机初始化CNN参数的方法,目前的训练数据量是不够的. 基于此，采用有监督的预训练，使用一个大的数据集（ImageNet )来训练AlexNet，得到一个1000分类的预训练（Pre-trained）模型。网络优化求解时采用随机梯度下降法（SGD），学习率大小为0.001.

3.fine-tuning阶段

接着采用 selective search 搜索出来的候选框（PASCAL VOC 数据库中的图片）继续对上面预训练的CNN模型进行fine-tuning训练。

将原来预训练模型最后的1000-way的全连接层（分类层）换成21-way的分类层（20类物体+背景），然后计算每个region proposal和ground truth 的IoU，对于IoU>0.5的region proposal被视为正样本，否则为负样本（即背景）。IOU（Intersection over Union），也称为Jaccard系数，是一种用于评估目标检测和图像分割任务性能的常用指标。它衡量预测边界框或分割掩码与真实边界框或分割掩码之间的重叠程度。

另外，由于对于一张图片的多有候选区域来说，负样本是远远大于正样本数，所以需要将正样本进行上采样来保证样本分布均衡。在每次迭代的过程中，选择层次采样，每个mini-batch中采样两张图像，从中随机选取32个正样本和96个负样本组成一个mini-batch（128，正负比：1：3）。我们使用0.001的学习率和SGD来进行训练。

4.提取并保存RP的特征向量
提取特征的CNN网络经过了预训练和微调后不再训练，就固定不变了，只单纯的作为一个提特征的工具了。虽然文中训练了CNN网络对region proposal进行分类，但是实际中，这个CNN的作用只是提取每个region proposal的feature。
所以，我们输入VOC训练数据集，SS搜索出2000个RP后输入进CNN进行前向传播，然后保存AlexNet的FC7层4096维的features，以供后续的SVM分类使用。

2.2.3 SVM分类

这是一个二分类问题，假设我们要检测车辆。我们知道只有当bounding box把整量车都包含在内，那才叫正样本；如果bounding box 没有包含到车辆，那么我们就可以把它当做负样本。但问题是当我们的检测窗口只有部分包含物体，那该怎么定义正负样本呢？作者测试了IOU阈值各种方案数值0,0.1,0.2,0.3,0.4,0.5。最后通过训练发现，如果选择IOU阈值为0.3效果最好（选择为0精度下降了4个百分点，选择0.5精度下降了5个百分点）,即当重叠度小于0.3的时候，我们就把它标注为负样本。一旦CNN f7层特征被提取出来，那么我们将为每个物体类训练一个svm分类器。当我们用CNN提取2000个候选框，可以得到20004096这样的特征向量矩阵，然后我们只需要把这样的一个矩阵与svm权值矩阵4096N点乘(N为分类类别数目，因为我们训练的N个svm，每个svm包含了4096个权值w)，就可以得到结果了。

得到的特征输入到SVM进行分类看看这个feature vector所对应的region proposal是需要的物体还是无关的实物(background) 。排序，canny边界检测之后就得到了我们需要的bounding-box。

总结一下：整个系统分为三个部分：1.产生不依赖与特定类别的region proposals，这些region proposals定义了一个整个检测器可以获得的候选目标2.一个大的卷积神经网络，对每个region产生一个固定长度的特征向量3.一系列特定类别的线性SVM分类器。

2.2.4 非极大值抑制法（NMS）

目的: 筛选候选区域，目标是一个物体只保留一个最优的框，来抑制那些冗余的候选框,一句话就是去掉多余的region proposals.

selective research会产生2k个region proposals。经过svm打分后，一个物体可能就有多个框。而我们只需要一个最优的（分最好的框），经过NMS抑制冗余的框，得到下图。

2.2.5边框位置修正 Boundingbox Regression

去掉了冗余的box还不够，因为这些box都是算法生成的，与ground truth(物体真实位置)肯定存在出入，我们希望这些box能尽可能接近ground truth。例如下图

红色框是ground truth，绿色框是生成的region proposal，这里看到我们需要修正绿色框，让它更接近红色框。于是使用了一组BBox Repression（每个分别对应一个类别）来做这个事情，输入原本的框，预测出一个新的框来接近ground truth。因为一个框可以用四个值来表示（x,y,w,h）分别是框的中心点横坐标，中心点纵坐标，宽，高。