单阶段目标检测方法SSD介绍与分析

SSD目标检测算法,完整详细讲解

  • SSD(single shot multi-box detector)
    • 1. 简介
    • 2. 模型结构
      • backbone
      • neck
      • head

SSD(single shot multi-box detector)

1. 简介

SSD是一种单阶段目标检测方法,如下图所示目标检测方法。
单阶段目标检测方法SSD介绍与分析_第1张图片
单阶段和双阶段的区别:

  1. 双阶段第一阶段, 主要是找出目标物体出现的位置,初步得到建议框,这一部分时间花费较长;双阶段第二阶段, 对建议框,进行精确位置回归,和物体的分类。
  2. 单阶段:不需要建议框,直接进行类别概率和位置坐标值的回归。

因此双阶段的优缺点就是位置更准确,准确率更高;但是时间更久。
单阶段的优缺点就是速度快,但是精确率会有损失。

2. 模型结构

单阶段目标检测方法SSD介绍与分析_第2张图片
单阶段目标检测方法SSD介绍与分析_第3张图片
原来的图更突出(上),我重新画了一个图(下),为了突出网络结构内容。

backbone

(1)位置
单阶段目标检测方法SSD介绍与分析_第4张图片

backbone就是背脊的意思,就是上图红框标出的部分。
(2)作用
这一部分的作用就是提取特征,后面预测框位置回归和分类回归,都是基于这一部分提取的特征进行的。
(3)详细信息(以VGG16为例)
这一部分其实就是一个深度卷积神经网络。原始的SSD是以VGG16的conv4_3层输出到预测部分,然后把conv5_3层特征图层输出到下一层。

neck

(1)位置
单阶段目标检测方法SSD介绍与分析_第5张图片
提取特征输出到预测部分之间的部分均可称为Neck。
(2)作用
主要是对提取的特征进行二次加工再给预测部分。这里说的二次加工,如FPN、PAN、深浅层特征融合等。
(3)详细信息
SSD在这一部分主要是对backbone输出的特征进行下采样(1×1的卷积对通道进行压缩,步长为2的3×3卷积进行下采样,相比maxpooling保留更多细节信息),并将各个下采样的特征图层输入到预测部分。

head

(1)位置
单阶段目标检测方法SSD介绍与分析_第6张图片
head就是最终的预测部分,输出预测结果。
(2)过程

  1. 对于每个预测框,首先根据置信度确定类别,并去除类别是背景的框;
  2. 去除类别置信度低于阈值(自己设定,如0.6)的框;
  3. 对剩下的预测框进行解码。根据预测的偏置(回归的是位置偏置,不是位置)与先验框位置大小得到位置参数(做clip,防止预测框位置超出图片);
  4. 对预测框根据置信度进行排列,保留前k个预测框;
  5. 最后就是进行NMS算法,过滤掉那些重叠度较大的预测框;
  6. 剩余的预测框就是检测结果。

(3)详细信息
直接对位置偏执和类别置信度进行回归,这就需要卷积出来的特征图层满足大小为 n×n×(a×(4+k))

  1. n×n就是特征图层每个通道上的大小,因为是以每个像素为锚框进行先验框的标定;
  2. (a×(4+k))就是通道量:
    (1)a就是预测框的数量;
    (2)4就是各个偏置 (x,y,w,h);
    (3)k就是需要预测的类别数量;

你可能感兴趣的:(目标检测,深度学习,计算机视觉)