深層学習に基づく物体検出アルゴリズムの詳細
左の本文を選ぶと、右側の官報原文画像で該当箇所を照合できます。
### 2.2.2.1 两阶段检测算法
两阶段检测算法首先生成候选区域(Region Proposals),然后对这些区域进行分类和边界框回归。典型的代表包括R-CNN系列算法(如Fast R-CNN、Faster R-CNN)。
### 2.2.2.2 单阶段检测算法
单阶段检测算法直接在图像上进行密集采样,同时完成目标分类和位置回归。这类算法速度更快,适合实时应用。典型代表有YOLO系列和SSD算法。
随着深度学习的发展,基于深度学习的目标检测算法逐渐取代了传统的目标检测算法。根据是否使用候选区域,可以将基于深度学习的目标检测算法分为两类:一类是基于候选区域的算法,另一类是不需要候选区域的算法。
(1)基于候选区域的目标检测算法
基于候选区域的目标检测算法首先通过选择性搜索等方法生成一系列候选区域,然后利用卷积神经网络对每个候选区域进行分类和回归,从而得到目标的类别和位置信息。这类算法的代表有R-CNN、Fast R-CNN、Faster R-CNN等。其中,R-CNN是第一个将深度学习应用于目标检测的算法,它通过选择性搜索生成候选区域,然后利用卷积神经网络提取特征,最后通过支持向量机进行分类和边界框回归。Fast R-CNN在R-CNN的基础上进行了改进,它将整个图像输入到卷积神经网络中,然后通过ROI池化层提取每个候选区域的特征,最后通过全连接层进行分类和边界框回归。Faster R-CNN在Fast R-CNN的基础上进一步改进,它引入了区域提议网络(Region Proposal Network, RPN),可以直接从卷积特征图中生成候选区域,从而提高了检测速度和精度。
(2)不需要候选区域的目标检测算法
不需要候选区域的目标检测算法直接将整个图像输入到卷积神经网络中,然后通过多个卷积层和池化层提取特征,最后通过全连接层或卷积层直接预测目标的类别和位置信息。这类算法的代表有YOLO、SSD等。其中,YOLO(You Only Look Once)是一种实时目标检测算法,它将整个图像划分为S×S个网格,每个网格负责预测B个边界框及其置信度和C个类别概率。SSD(Single Shot MultiBox Detector)是一种多尺度目标检测算法,它在不同尺度的特征图上进行预测,从而能够检测到不同大小的目标。