当前位置:   article > 正文

目标检测系列模型发展历程_目标检测发展史

目标检测发展史

常见数据集:

VOC-->COCO

VOC数据集

特点

  • 发布于2005-2012年,由多个项目(如VOC2007和VOC2012)组成。
  • 包含20个物体类别,涵盖动物、交通工具和日常物品等。
  • 提供了物体检测、物体分类、物体分割等任务的标注。
  • 图像数量相对较少,但提供了高质量的标注。

COCO数据集

特点

  • 于2014年发布,目的是推动场景理解技术的发展。
  • 包含80个物体类别,涵盖了更多的日常物品和场景。
  • 提供了丰富的标注,包括物体检测、关键点检测、语义分割和全景分割等。
  • 包含超过200,000张图像,每张图像平均标注了3.5个不同的物体,图像中物体的上下文信息丰富。

模型发展:

RCNN-->Fast RCNN-->Faster RCNN-->Mask RCNN

这一系列的模型(RCNN、Fast RCNN、Faster RCNN、Mask RCNN)代表了计算机视觉特别是在物体检测和分割领域的一系列重大进展。下面,我会概述每个模型的特点、网络结构、优缺点,以及它们的发展历史。

1. RCNN(Regions with CNN features)

- 特点与网络结构:RCNN(Regions with Convolutional Neural Networks)通过结合区域提取(Region Proposal)技术和卷积神经网络(CNN)来检测图片中的物体。首先,使用选择性搜索(Selective Search)算法从图片中提取约2000个候选区域,然后将这些区域调整到固定大小,并用CNN提取特征,最后通过SVM(支持向量机)分类器对每个区域进行分类
- 优点:提出了使用深度学习进行物体检测的新范式,大幅提升了检测精度。
- 缺点:速度慢,因为对每个候选区域独立运行CNN是非常耗时的;训练过程繁琐,包括预训练CNN、训练SVM分类器和边界框回归器。

RCNN(Regions with CNN features)算法是一种结合了区域提案(Region Proposal)卷积神经网络(CNN)的物体检测方法。其核心步骤可以总结如下:

  1. 区域提案:首先,使用选择性搜索(Selective Search)算法在输入图像上生成大约2000个候选区域(也称为区域提案)。这些区域是图像中可能包含物体的部分。

  2. 特征提取:对于每个区域提案,将其缩放到固定大小(通常是227x227像素),然后通过预训练的卷积神经网络(如AlexNet、VGG等)运行,以提取该区域的特征。这一步骤是独立对每个区域进行的,因此是计算密集型的。

  3. 分类:提取的特征被送入一系列的支持向量机(SVM)分类器中,每个SVM分类器负责判断区域是否属于某个特定的类别(例如人、车等)。因此,对于每个类别,都有一个单独的SVM模型。

  4. 边界框回归:对于被分类器认为含有物体的区域,使用线性回归模型对其边界框(bounding box)的位置进行微调,以更精确地定位物体。这一步骤有助于改进最终的检测精度。

  5. 非极大值抑制(NMS):对于每个类别,可能有多个区域提案被识别为包含该类别的物体,这些区域之间可能高度重叠。非极大值抑制是一种技术,用于在每个类别中只保留最佳的区域提案,即抑制(删除)所有与最高得分的区域提案重叠度较高的其他区域提案,从而减少重复检测。

RCNN通过这些步骤实现了当时领先的检测性能,但也因为需要对每个区域提案独立运行CNN,导致其处理速度较慢,不适合实时应用。此外,其训练过程相对复杂,需要分别训练CNN、SVM和边界框回归模型。

2. Fast RCNN

- 特点与网络结构:为了解决RCNN速度慢的问题,Fast RCNN在整张图片上运行CNN,一次性提取特征图,然后对于每个候选区域从这个共享的特征图上裁剪出特征(使用ROI池化层),之后再通过一系列的全连接层进行分类和边界框回归。
- 优点:相比于RCNN,大大提升了速度和效率,因为它在整张图上只运行一次CNN。
- 缺点:仍依赖于慢速的选择性搜索算法来生成区域提案。

Fast RCNN是对RCNN的一种改进,旨在解决RCNN处理速度慢和训练过程复杂的问题。它通过共享卷积计算和使用多任务损失函数来同时进行分类和边界框回归,显著提高了处理速度和训练效率。下面是Fast RCNN的核心步骤:

  1. 整图卷积特征提取:与RCNN不同,Fast RCNN首先对整张输入图像运行一次卷积神经网络,生成整个图像的卷积特征图。这一步骤大大减少了重复的计算,因为它避免了对每个区域提案分别提取特征。

  2. 区域提案:与RCNN类似,Fast RCNN仍然使用选择性搜索或其他方法来生成区域提案,但这些区域提案是直接在卷积特征图上操作,而不是原始图像上。

  3. RoI(Region of Interest)池化:对于每个区域提案,Fast RCNN从卷积特征图上裁剪对应的特征区域,并使用RoI池化层将这些特征区域调整为固定大小(例如7x7),以便进行进一步处理。这一步是通过在卷积特征图上操作实现的,从而保持了计算效率。

  4. 分类与边界框回归:RoI池化后的特征被送入一系列全连接层,最后分叉成两个输出层:一个用于物体分类,另一个用于边界框回归。这里使用的是多任务损失,即同时优化分类损失和边界框回归损失。

  5. 非极大值抑制(NMS):和RCNN一样,Fast RCNN也使用非极大值抑制来处理每个类别的多个检测,保留最佳的检测结果,从而减少重复检测。

Fast RCNN的这些改进大幅提高了物体检测的速度和训练效率,同时简化了训练过程,因为整个模型可以通过反向传播算法端到端地训练。然而,虽然Fast RCNN在处理速度上有显著改进,但它仍依赖于传统的区域提案方法(如选择性搜索),这部分仍是速度的瓶颈。

3. Faster RCNN

- 特点与网络结构:为了克服依赖慢速选择性搜索算法的缺点,Faster RCNN引入了区域提案网络(RPN),它是一个全卷积网络,可以在CNN特征图上实时生成区域提案。这些提案随后用于ROI池化和后续的分类及边界框回归。
- 优点:实现了端到端的训练和显著的速度提升,同时保持了高精度。
- 缺点:虽然速度有所提升,但在处理非常高分辨率的图像时,速度和效率仍有限。

Faster RCNN进一步改进了Fast RCNN,引入了区域提案网络(RPN,Region Proposal Network),以解决依赖慢速选择性搜索生成区域提案的问题,实现了几乎实时的物体检测。以下是Faster RCNN算法的核心步骤:

  1. 共享卷积层:与Fast RCNN类似,Faster RCNN首先对整张图像使用卷积神经网络(CNN)生成特征图。这一步骤不仅用于后续的物体检测,也为区域提案网络(RPN)提供了输入特征。

  2. 区域提案网络(RPN)

    • RPN是Faster RCNN的核心创新之一。它在特征图上滑动一个小网络,该网络在每个位置同时预测多个区域提案及其“对象得分”(objectness score)。
    • RPN使用一组预定义大小和比例的锚点框(anchor boxes)作为参考,通过学习调整这些锚点框的位置和大小,生成精确的区域提案。
    • 对于每个锚点框,RPN预测两类得分(是对象还是非对象)和四个调整值(用于精确定位边界框)。
  3. RoI池化

    • 使用RPN生成的区域提案,Faster RCNN从共享的卷积特征图中裁剪出对应的特征区域。
    • 通过RoI池化层将这些特征区域调整为固定大小,以便进一步处理。
  4. 分类与边界框回归

    • 和Fast RCNN一样,Faster RCNN将RoI池化后的特征送入一系列全连接层,这些层分叉成两个输出:一个用于物体分类,另一个用于边界框回归。
    • 通过多任务损失同时优化分类和边界框定位,实现更精确的检测。
  5. 非极大值抑制(NMS)

    • 对于每个类别,可能会有多个重叠的检测框。通过非极大值抑制,Faster RCNN去除重叠度较高的检测框,只保留得分最高的框,从而减少重复检测。

Faster RCNN的这些改进显著提升了检测的速度和精度,使其成为深度学习物体检测领域的一个里程碑。RPN的引入使得模型可以端到端地进行训练,同时生成高质量的区域提案,大大减少了对外部区域提案方法的依赖。

4. Mask RCNN

- 特点与网络结构:Mask RCNN在Faster RCNN的基础上增加了一个分支,用于生成物体的像素级掩码,实现了物体检测和实例分割的同时进行。它使用ROIAlign代替了ROIPooling,解决了ROI池化因为量化而造成的精度损失问题。
- 优点:能够同时进行高精度的物体检测和实例分割;ROIAlign改进提高了分割精度。
- 缺点:相比于仅进行物体检测的模型,计算成本更高,速度更慢。

Mask RCNN是在Faster RCNN的基础上发展起来的,它不仅能进行物体检测,还能进行实例分割,即对每个检测到的物体,Mask RCNN能够生成一个精确的像素级掩码,用以区分物体的具体轮廓。以下是Mask RCNN算法的主要步骤:

  1. 卷积特征提取:Mask RCNN首先使用卷积神经网络(通常是ResNet加上FPN,Feature Pyramid Network)处理整张图像,生成多尺度的特征图。这些特征图既用于后续的区域提案,也用于生成掩码。

  2. 区域提案网络(RPN):通过在特征图上运行RPN生成区域提案。RPN通过学习在每个位置预测物体边界框和对象得分,产生高质量的区域提案。

  3. RoIAlign:与Faster RCNN中的RoIPooling不同,Mask RCNN使用RoIAlign来从特征图上精确裁剪和调整每个提案区域到固定大小,解决了RoIPooling因量化而引入的轻微位置偏差,提高了掩码预测的精度。

  4. 分类、边界框回归与掩码预测

    • 对于每个RoIAlign后的区域,Mask RCNN使用全连接层进行分类和边界框回归。
    • 同时,对每个区域使用一个小的全卷积网络(FCN)生成对应的物体掩码。不同于分类和边界框回归,掩码预测是在每个类别上独立进行的,意味着每个类别都会有一个掩码预测,最后根据分类结果选择相应的掩码。
  5. 非极大值抑制(NMS):和Faster RCNN一样,Mask RCNN也应用NMS来去除同一类别中重叠度较高的检测,确保每个实例只被检测一次。

  6. 多任务损失:Mask RCNN通过一个多任务损失函数来同时优化分类、边界框回归和掩码预测。这个损失函数包括分类损失、边界框损失和掩码损失,它们共同指导模型的训练。

Mask RCNN通过这些步骤,能够实现同时对物体进行高精度的检测和像素级的实例分割,显著提升了实例分割的准确度和效率。其引入的RoIAlign技术特别解决了实例分割中的对齐问题,是其能够成功应用于实例分割的关键所在。

SSD-->FSSD-->DSSD

SSD(Single Shot MultiBox Detector),FSSD(Feature Fusion Single Shot Multibox Detector),和DSSD(Deconvolutional Single Shot Detector)是一系列在物体检测领域中发展起来的深度学习模型。它们主要致力于提高检测的速度和准确率。下面将依次介绍这些模型的特点、网络结构、优缺点以及它们的发展历史。

5. SSD(Single Shot MultiBox Detector)

  • 特点与网络结构SSD是一种单次检测模型,能够在不牺牲准确率的情况下提供快速的检测速度。它通过在多个卷积特征图上预测物体的存在和类别,可以同时处理多种尺度的物体检测。SSD网络包括一个基础的特征提取网络(通常是预训练的VGG16或ResNet),后接多个卷积层,这些层在不同深度上产生不同尺度的特征图,用于检测不同大小的物体。
  • 优点:高效且实现简单,能够在单个网络中同时处理不同尺度的物体检测。
  • 缺点:在小物体检测上的表现不如大物体,因为深层的特征图分辨率较低,丢失了大量细节信息。

SSD(Single Shot MultiBox Detector)算法是一种流行的物体检测方法,它能够在单次前向传播中同时完成物体检测的定位和分类,因此速度快且效率高。下面是SSD算法的主要步骤:

  1. 基础网络层:SSD首先使用一个预训练的网络(如VGG16、ResNet等)作为基础特征提取网络,去掉了原网络的全连接层。这个基础网络负责从输入图像中提取低层次的特征。

  2. 额外的特征层:在基础网络之后,SSD添加了一系列额外的卷积层,这些层逐渐减小尺寸,用于检测不同大小的物体。通过在这些不同层级的特征图上进行检测,SSD可以有效地覆盖各种尺度的物体。

  3. 锚点框(Anchors):在每个特征图上,SSD定义了一系列的锚点框(也称为先验框或默认框)。这些锚点框以不同的比例(aspect ratios)和尺寸(scales)覆盖了整个图像,用于匹配不同大小和形状的物体。

  4. 定位和分类预测:对于每个特征图上的每个锚点框,SSD使用一组小型卷积滤波器来预测该锚点框相对于匹配的真实物体边界框的偏移量,以及锚点框包含每个类别物体的概率。

  5. 损失计算:SSD采用了一个多任务损失函数,结合了定位损失(例如Smooth L1损失)和分类损失(例如交叉熵损失),用于在训练过程中同时优化模型的定位和分类性能。

  6. 非极大值抑制(NMS):由于每个锚点框可能预测多个物体,SSD通过非极大值抑制来筛选出每个类别中得分最高的检测结果,以此减少检测到的重复物体。

6. FSSD(Feature Fusion Single Shot Multibox Detector)

  • 特点与网络结构:FSSD是对SSD的改进,通过增加特征融合模块来强化模型对小物体的检测能力。它利用较浅层的高分辨率特征图与深层的低分辨率特征图进行融合,以此来丰富特征信息,改善对小物体的检测性能。
  • 优点:通过特征融合,FSSD在保持快速检测速度的同时,提高了对小物体的检测精度。
  • 缺点:特征融合可能会增加计算负担,稍微影响模型的运行速度。

FSSD(Feature Fusion Single Shot MultiBox Detector)算法是对SSD(Single Shot MultiBox Detector)的改进,旨在通过特征融合提高对小物体检测的准确性。FSSD利用来自基础网络不同层的特征,并通过融合这些特征来增强模型的表现。以下是FSSD算法的主要步骤:

  1. 基础网络和额外特征层:FSSD同样使用一个预训练的网络(如VGG16、ResNet等)作为其基础特征提取网络,并在其后添加额外的卷积层以产生多尺度的特征图。这与SSD的做法类似,目的是为了覆盖不同大小的物体。

  2. 特征融合模块

    • FSSD的关键改进在于引入了特征融合模块,该模块结合了不同层级的特征图。
    • 从基础网络和额外卷积层中选取不同深度的特征图,通过上采样或下采样使它们具有相同的空间尺寸,然后将这些特征图融合在一起,通常是通过相加或连接操作。
    • 融合后的特征图包含了丰富的上下文信息,有助于提高对小物体的检测能力。
  3. 锚点框(Anchors)的定义:在融合的特征图上定义锚点框,与SSD相似,这些锚点框覆盖了不同的比例和尺度,用于匹配图像中的物体。

  4. 定位和分类预测:对于每个锚点框,FSSD利用融合后的特征图进行定位和分类预测。这包括预测锚点框相对于匹配的真实物体边界框的偏移量,以及锚点框内物体属于每个类别的概率。

  5. 损失计算:FSSD同样采用多任务损失函数来优化定位和分类的性能,通常结合定位损失(如Smooth L1损失)和分类损失(如交叉熵损失)。

  6. 非极大值抑制(NMS):使用非极大值抑制来筛选出每个类别中得分最高的检测结果,减少重复检测。

7. DSSD(Deconvolutional Single Shot Detector)

  • 特点与网络结构:DSSD在SSD的基础上引入了反卷积(Deconvolution)层,用于从低分辨率特征图中恢复出更多的细节信息,进一步改善对小物体的检测能力。DSSD通过结合深层和浅层特征,在提升检测精度的同时尽量减少速度的损失。
  • 优点:增强了模型对于小物体以及细节特征的识别能力,提高了检测的整体精度。
  • 缺点:引入的反卷积层会增加模型的计算复杂度,可能导致检测速度下降。

DSSD(Deconvolutional Single Shot Detector)是在SSD(Single Shot MultiBox Detector)基础上进行改进的物体检测算法。DSSD通过引入反卷积(deconvolution)层来增加特征图的分辨率,目的是改善对小物体的检测性能同时保持对大物体检测的准确性。下面是DSSD算法的主要步骤:

  1. 基础网络层:DSSD使用一个预训练的深度卷积网络(如ResNet)作为其基础特征提取网络。这个基础网络负责从输入图像中提取丰富的特征。

  2. 额外的特征层:在基础网络之后,DSSD增加了额外的卷积层来产生不同尺度的特征图,这些特征图用于后续的物体检测。这一步骤与SSD相似,旨在捕捉不同大小物体的特征。

  3. 反卷积层

    • 为了提高特征图的分辨率,DSSD在额外的特征层之后引入了反卷积(deconvolution)层。反卷积层通过学习将低分辨率的特征图上采样,从而恢复更多的空间信息。
    • 这些高分辨率的特征图有助于改善对小物体的检测性能,因为它们包含了更多的细节信息。
  4. 特征融合

    • DSSD利用从基础网络和反卷积层获得的特征图进行特征融合。这包括将深层的、高级语义信息丰富的特征与浅层的、高分辨率的特征相结合,以增强模型对不同尺寸物体的检测能力。
    • 特征融合通常通过元素相加或特征拼接等方法实现。
  5. 锚点框(Anchors)的定义:在融合后的特征图上定义锚点框,这些锚点框以不同的比例和尺度覆盖图像,用于捕捉各种大小和形状的物体。

  6. 定位和分类预测:对于每个锚点框,DSSD利用融合特征图进行定位和分类预测,包括预测边界框的偏移量和物体的类别概率。

  7. 损失计算:DSSD使用多任务损失函数来同时优化定位和分类任务,包括定位损失(如Smooth L1损失)和分类损失(如交叉熵损失)。

  8. 非极大值抑制(NMS):应用非极大值抑制来筛选出每个类别中得分最高的检测框,减少检测结果中的重复物体。

YOLO系列:YOLO发展史

指标评估:

目标检测评估指标

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/我家自动化/article/detail/559285
推荐阅读