文章目录

  • 一、SSD((Single Shot MultiBox Detector))
    • 1.Default box
    • 2.训练过程
    • 3.数据增强
    • 4.其他变体
      • (1)DSSD
      • (2)DSOD
      • (3)FSSD
      • (4)RSSD
  • 二、YOLO(You Only Look Once)
    • 1.YOLO_V1
    • 2.YOLO_V2、YOLO9000
      • (1)better
      • (2)faster
      • (3)Stronger
    • 3.YOLO_V3
      • (1)backbone
      • (2)output
      • (3)损失函数
    • 4.YOLO_V4
      • (1)输入端
        • Mosaic数据增强
      • (2)BackBone
        • CSPDarknet53
        • Mish激活函数
        • Dropblock
      • (3)Neck
        • SPP模块
        • FPN+PAN
      • (4)Prediction
        • CIOU_loss
        • DIOU_nms
    • 5.YOLO_V5
      • (1)输入端
        • 自适应锚框计算
        • 自适应图片缩放
      • (2)Backbone
        • Focus结构
        • CSP结构
      • (3)Neck:FPN+PAN
      • (4)输出端
      • (5)四种网络对比
        • 深度对比
        • 宽度对比
  • 三、RetinaNet
  • 四、Anchor-free模型

参考文章:
论文笔记-2019-Object Detection in 20 Years: A Survey
基于深度学习的目标检测算法综述(一)
基于深度学习的目标检测算法综述(二)
基于深度学习的目标检测算法综述(三)

优秀的GitHub仓库:
deep learning object detection
Awesome Object Detection

比较全面的整理:
我这两年的目标检测
最全目标检测相关资料整理
我这两年收藏的目标检测好文分享

一、SSD((Single Shot MultiBox Detector))

SSD算法在传统的基础网络(比如VGG)后添加了5个特征图尺寸依次减小的卷积层,对5个特征图的输入分别采用2个不同的3*3的卷积核进行卷积,一个输出类别的confidence;一个输出回归用的localization,每个default box生成4个坐标值,最后将5个特征图上的结果合并(Contact),送入loss层。

SSD算法的结构图如下:

1.Default box

SSD算法的核心是Default box层,Default box层利用了Anchor的机制,如下图所示,这里假定有 8×8 和 4×4 两种不同尺度的 feature map。第一个概念是feature map cell,feature map cell 是指 feature map 中每一个小格子,如图中分别有 64 和 16 个 cell。另一个概念是default box,是指在feature map的每个小格(cell)上都有一系列固定大小的box
假设每个 feature map cell 有 k 个 default box,那么对于每个default box都需要预测 c 个类别 score 和 4 个 offset,那么如果一个 feature map的大小是 m×n,也就是有 m * n 个 feature map cell,那么这个 feature map就一共有(c+4)* k * m * n 个输出。这些输出个数的含义是:采用3×3的卷积核对该层的feature map卷积时卷积核的个数,包含两部分:数量 c * k * m * n 是 confidence 输出,表示每个 default box 的 是某一类别的 confidence;数量4 * k * m * n是 localization 输出,表示每个default box 回归后的坐标。

对于先验框的尺度,其遵守一个线性递增规则:随着特征图大小降低,先验框尺度线性增加:

这里smin是0.2,表示最底层的尺度是0.2,;smax是0.9,表示最高层的尺度是0.9。通过这种计算方法,可以得出6个特征图的default box尺度分别为:[30,60,111,162,213,264]。长宽比用??表示为下式(注意这里一共有5种长宽比):

通过下面的公式计算 default box 的宽度w和高度h:

另外当 aspect ratio 为1时,作者还增加一种 scale 的 default box:

因此,对于每个 feature map cell 而言,一共有 6 种 default box。

2.训练过程

正样本获得:将得到的 default box 和真实框进行 IOU 匹配,大于0.5即认为是正样本

负样本获得:很显然负样本的数量远多于正样本,因此先将每一个物体位置上对应 predictions(default boxes)是 negative 的 boxes 进行排序,按照 default boxes 的 confidence 的大小。 选择最高的几个,最终使正负样例比例为1:3。

损失函数:定义为位置误差(locatization loss, loc)与置信度误差(confidence loss, conf)的加权和:

  • N 是与 ground truth box 相匹配的 default boxes 个数
  • localization loss(loc) 是 Fast R-CNN 中 Smooth L1 Loss,用在 predict box(l) 与 ground truth box(g) 参数(即中心坐标位置,width、height)中,回归 bounding boxes 的中心位置,以及 width、height
  • confidence loss(conf) 是 Softmax Loss,输入为每一类的置信度 c
  • 权重项 α,设置为 1

3.数据增强

每一张训练图像,随机的进行如下几种选择:

  • 使用原始的图像
  • 随机采样多个 patch,与物体之间最小的 jaccard overlap 为:0.1,0.3,0.5,0.7 与 0.9

采样的 patch 是原始图像大小比例是 [0.1,1],aspect ratio 在 12 与 2 之间。当 groundtruth box 的 中心(center)在采样的 patch 中时,我们保留重叠部分。在这些采样步骤之后,每一个采样的 patch 被 resize 到固定的大小,并且以 0.5 的概率随机的 水平翻转(horizontally flipped)

4.其他变体

对于 SSD 的改进可以包括对主干网络的改进,例如使用 ResNet 等效果更好的网络作为主干网络,还可以对预测网络进行改进。

(1)DSSD

SSD 算法对小目标不够鲁棒的原因是浅层特征层的表征能力不够强,因此 DSSD使用ResNet-101代替VGG作为主干网络,在‘SSD layers‘后面添加了反卷积层和feed-forward连接,然后使用residual block作为预测层。。

(2)DSOD

SSD+DenseNet,可以不使用预训练模型进行训练

(3)FSSD

借鉴FPN的思想,重构了一组金字塔特征

(4)RSSD

采用rainbow concatenation形式(池化+反卷积)融合不同层的特征,增加不同特征层之间特征图的关系和个数

二、YOLO(You Only Look Once)

YOLO的核心思想就是利用整张图作为网络的输入,直接在输出层回归bounding box的位置和bounding box所属的类别。

1.YOLO_V1

YOLO的实现方法为:

  • 将一幅图像分成SxS个网格(grid cell),如果某个object的中心 落在这个网格中,则这个网格就负责预测这个object。
  • 每个网格要预测B个bounding box,每个bounding box除了要回归自身的位置之外,还要附带预测一个confidence值。如果grid cell里面没有object,confidence就是0,如果有,则confidence score等于预测的box和ground truth的IOU值。
  • 每个bounding box要预测(x, y, w, h)和confidence共5个值,每个网格还要预测一个类别信息,记为C类。则SxS个网格,每个网格要预测B个bounding box还要预测C个categories。输出就是S x S x (5*B+C)的一个tensor。

YOLO_V1中取S=7,B=2,C=20(因为PASCAL VOC有20个类别),所以最后有7 * 7 * 30个tensor,下图为YOLO_V1结构图

关于损失函数的设计,前面两行表示localization error(即坐标误差),第一行是box中心坐标(x,y)的预测,第二行为宽和高的预测。这里注意用宽和高的开根号代替原来的宽和高,这样做主要是因为相同的宽和高误差对于小的目标精度影响比大的目标要大。

第三、四行表示bounding box的confidence损失,就像前面所说的,分成grid cell包含与不包含object两种情况。这里注意下因为每个grid cell包含两个bounding box,所以只有当ground truth 和该网格中的某个bounding box的IOU值最大的时候,才计算这项。

第五行表示预测类别的误差,注意前面的系数只有在grid cell包含object的时候才为1。

在测试的时候,每个网格预测的 class 信息和 bounding box 预测的 confidence信息相乘,就得到每个 bounding box 的 class-specific confidence score

等式左边第一项就是每个网格预测的类别信息,第二、三项就是每个 bounding box 预测的 confidence。这个乘积即 encode 了预测的 box 属于某一类的概率,也有该 box 准确度的信息。得到每个 box 的 class-specific confidence score 以后,设置阈值,滤掉得分低的 boxes,对保留的 boxes 进行 NMS 处理,就得到最终的检测结果。

2.YOLO_V2、YOLO9000

YOLO_v2是YOLO的升级版,YOLO9000的主要检测网络也是YOLO v2,同时对数据集做了融合,使得模型可以检测9000多类物体。而提出YOLO9000的原因主要是目前检测的数据集数据量较小,因此利用数量较大的分类数据集来帮助训练检测模型。YOLO_V2、YOLO9000对比YOLO_v1主要有better、faster、stronger三个方面

(1)better

  • Batch Normalization:在YOLOv2中作者为每个卷积层都添加了BN层,另外加入BN后就把dropout去掉了。实验证明添加了BN层可以提高2%的mAP。
  • High Resolution Classifier:原来的YOLO网络在预训练的时候采用的是224 * 224的输入(这是因为一般预训练的分类模型都是在ImageNet数据集上进行的),然后在detection的时候采用448 * 448的输入,这会导致从分类模型切换到检测模型的时候,模型还要适应图像分辨率的改变。而YOLOv2则将预训练分成两步:先用224 * 224的输入从头开始训练网络,大概160个epoch,然后再将输入调整到448 * 448,再训练10个epoch。这两步都是在ImageNet数据集上操作。最后detection的时候用448 * 448的图像作为输入就可以顺利过渡了。作者的实验表明这样可以提高几乎4%的MAP。
  • Convolutional With Anchor Boxes:YOLOv2首先将原网络的全连接层和最后一个pooling层去掉,使得最后的卷积层可以有更高分辨率的特征;然后缩减网络,用416*416大小的输入代替原来448 * 448。网络最终将416 * 416的输入变成13 * 13大小的feature map输出,也就是缩小比例为32。另外YOLOv2通过引入anchor boxes,预测的box数量超过了1千(以输出feature map大小为13 * 13为例,每个grid cell有9个anchor box的话,一共就是13 * 13 * 9=1521个)。实验证明:虽然加入anchor使得MAP值下降了一点(69.5降到69.2),但是提高了recall(81%提高到88%)。
  • Dimension Clusters:作者采用k-means的方式对训练集的bounding boxes做聚类,试图找到合适的anchor box。 通过IOU定义了如下的距离函数,使得误差和box的大小无关:
  • Direct Location prediction:用 Anchor Box 的方法,会让 model 变得不稳定,尤其是在最开始的几次迭代的时候。大多数不稳定因素产生自预测 Box 的(x,y)位置的时候。在基于region proposal的目标检测算法中,是通过预测tx和ty来得到(x,y)值,也就是预测的是offsets。

    YOLOv2中没有采用这种预测方式,而是沿用了YOLOv1的方法,就是预测边界框中心点相对于对应cell左上角位置的相对偏移值。为了将bounding box的中心点约束在当前cell中,使用sigmoid函数将tx、ty归一化处理,将值约束在0~1,这使得模型训练更稳定。
  • Fine-Grained Features :这里主要是添加了一个层:passthrough layer。这个层的作用就是将前面一层的26 * 26的feature map和本层的13 * 13的feature map进行连接,有点像ResNet。这样做的原因在于虽然13 * 13的feature map对于预测大的object以及足够了,但是对于预测小的object就不一定有效,所以通过合并前一层的size大一点的feature map,可以有效检测小的object。
  • Multi-Scale Training:YOLOv2中只有卷积层和池化层,因此不需要固定的输入图片的大小。为了让模型更有鲁棒性,作者引入了多尺度训练。就是在训练过程中,每迭代一定的次数,改变模型的输入图片大小。注意:这一步是在检测数据集上fine-tuning时候采用的,不要跟前面在Imagenet数据集上的两步预训练分类模型混淆。网络输入是416×416,经过5次max pooling之后会输出13×13的feature map,也就是下采样32倍,因此作者采用32的倍数作为输入的size,具体采用320、352、384、416、448、480、512、544、576、608共10种size。每次改变输入图片大小还需要对最后检测层进行处理,然后开始训练。这种网络训练方式使得相同网络可以对不同分辨率的图像做检测,也可以在准确率和速度上都取得一个不错的平衡。

(2)faster

  • Darknet-19:网络包含19个卷积层和5个max pooling层,整体上卷积卷积操作比YOLOv1中用的GoogleNet要少。最后用average pooling层代替全连接层进行预测。
  • 训练过程:
  1. 从头开始训练Darknet-19,数据集是ImageNet,训练160个epoch,输入图像的大小是224*224,初始学习率为0.1。另外在训练的时候采用了标准的数据增加方式比如随机裁剪,旋转以及色度,亮度的调整等。
  2. fine-tuning 网络,这时候采用448*448的输入,参数的除了epoch和learning rate改变外,其他都没变,这里learning rate改为0.001,并训练10个epoch。
  3. 把网络移植到detection,并开始基于检测的数据再进行fine-tuning。首先把最后一个卷积层去掉,然后添加3个3 * 3的卷积层,每个卷积层有1024个filter,而且每个后面都连接一个1 * 1的卷积层,1 * 1卷积的filter个数根据需要检测的类来定。

(3)Stronger

YOLO9000主要通过结合分类和检测数据集使得训练得到的检测模型可以检测约9000类物体。一方面要构造数据集,另一方面要解决模型训练问题,前者采用WordTree解决,后者采用Joint classification and detection。

根据各个类别之间的从属关系(根据WordNet)建立一种树结构WordTree,结合COCO和ImageNet建立的WordTree如下图所示:

WordTree中的根节点为"physical object",每个节点的子节点都属于同一子类,可以对它们进行softmax处理。在给出某个类别的预测概率时,需要找到其所在的位置,遍历这个path,然后计算path上各个节点的概率之积。

Joint classification and detection:作者的目的是训练一个 Extremely Large Scale 检测器。所以训练的时候使用 WordTree 混合了 COCO 检测数据集与 ImageNet 中的 Top9000 类,混合后的数据集对应的 WordTree 有 9418 个类。另一方面,由于 ImageNet 数据集太大了,作者为了平衡一下两个数据集之间的数据量,通过过采样(oversampling) COCO 数据集中的数据,使 COCO 数据集与 ImageNet 数据集之间的数据量比例达到 1:4。YOLO9000 的训练基于 YOLO v2 的构架,但是使用 3 priors 而不是 5 来限制输出的大小。当网络遇到检测数据集中的图片时则正常地反方向传播,当遇到分类数据集图片的时候,只使用分类的 loss 功能进行反向传播。同时作者假设 IOU 最少为 0.3。最后根据这些假设进行反向传播。

使用联合训练法,YOLO9000 使用 COCO 检测数据集学习检测图片中的物体的位置,使用 ImageNet 分类数据集学习如何对大量的类别中进行分类。

3.YOLO_V3

参考文章:yolo系列之yolo v3【深度解析】

YOLO_V3的结构如下:

这里推荐的模型结构可视化工具是:Netron

整个yolo_v3_body包含252层,组成如下:

(1)backbone

整个v3结构里面,是没有池化层和全连接层的。前向传播过程中,张量的尺寸变换是通过改变卷积核的步长来实现的,比如stride=(2, 2),这就等于将图像边长缩小了一半。yolo_v3也和v2一样,backbone都会将输出特征图缩小到输入的1/32。所以,通常都要求输入图片是32的倍数。

darknet-19是不存在残差结构,和VGG是同类型的backbone,而darknet-53是从resnet上借鉴过来的backbone。

darknet-19在速度上仍然占据很大的优势。可以看出yolo_v3并没有那么追求速度,而是在保证实时性(fps>36)的基础上追求performance。不过要想更快,还有一个 tiny-darknet作为backbone可以替代darknet-53。

(2)output

  • yolo v3借鉴了FPN采用多尺度来对不同size的目标进行检测的方法,输出了3个不同尺度的feature map,如上图所示的y1, y2, y3。

  • 对于COCO类别而言,有80个种类,yolo v3中每个网格单元预测3个box,每个box需要有(x, y, w, h, confidence)五个基本参数,然后还要有80个类别的概率。所以3*(5 + 80) = 255。

  • 对于yolo v3而言,对于先验框的选择:选用的b-box priors 的k=9,对于tiny-yolo的话,k=6。priors都是在数据集上聚类得来的,有确定的数值,9个anchor会被三个输出张量平分的。根据大中小三种size各自取自己的anchor。

  • v3对b-box进行预测的时候,采用了logistic regression。v3每次对b-box进行predict时,输出和v2一样是偏差,然后通过公式计算出绝对的(x, y, w, h, c)。logistic回归用于对anchor包围的部分进行一个目标性评分(objectness score),即这块位置是目标的可能性有多大。这一步是在predict之前进行的,可以去掉不必要anchor,可以减少计算量。

(3)损失函数

除了w, h的损失函数依然采用总方误差之外,其他部分的损失函数用的是二值交叉熵。

keras代码如下:

xy_loss = object_mask * box_loss_scale * K.binary_crossentropy(raw_true_xy, raw_pred[..., 0:2],from_logits=True)
wh_loss = object_mask * box_loss_scale * 0.5 * K.square(raw_true_wh - raw_pred[..., 2:4])
confidence_loss = object_mask * K.binary_crossentropy(object_mask, raw_pred[..., 4:5], from_logits=True) + \(1 - object_mask) * K.binary_crossentropy(object_mask, raw_pred[..., 4:5],from_logits=True) * ignore_mask
class_loss = object_mask * K.binary_crossentropy(true_class_probs, raw_pred[..., 5:], from_logits=True)xy_loss = K.sum(xy_loss) / mf
wh_loss = K.sum(wh_loss) / mf
confidence_loss = K.sum(confidence_loss) / mf
class_loss = K.sum(class_loss) / mf
loss += xy_loss + wh_loss + confidence_loss + class_loss

4.YOLO_V4

参考文章:深入浅出Yolo系列之Yolov3&Yolov4&Yolov5核心基础知识完整讲解
参考代码:Pytorch-YOLOv4

YOLO_V4结构图如下:

Yolov4的整体架构和Yolov3是相同的,不过使用各种新的算法思想对各个子结构都进行了改进。

Yolov4的五个基本组件:

  • CBM:Yolov4网络结构中的最小组件,由Conv+Bn+Mish激活函数三者组成。
  • CBL:由Conv+Bn+Leaky_relu激活函数三者组成。
  • Res unit:借鉴Resnet网络中的残差结构,让网络可以构建的更深。
  • CSPX:借鉴CSPNet网络结构,由卷积层和X个Res unint模块Concate组成。
  • SPP:采用1×1,5×5,9×9,13×13的最大池化的方式,进行多尺度融合。

主要从4个部分对YoloV4的创新之处进行解释:

(1)输入端

Yolov4对训练时的输入端进行改进,使得训练在单张GPU上也能有不错的成绩。比如数据增强Mosaic、cmBN、SAT自对抗训练。

Mosaic数据增强

Yolov4中使用的Mosaic是参考2019年底提出的CutMix数据增强的方式,但CutMix只使用了两张图片进行拼接,而Mosaic数据增强则采用了4张图片,随机缩放、随机裁剪、随机排布的方式进行拼接。

Mosaic数据增强主要有几个优点:

  • 丰富数据集:随机使用4张图片,随机缩放,再随机分布进行拼接,大大丰富了检测数据集,特别是随机缩放增加了很多小目标,让网络的鲁棒性更好。
  • 减少GPU:可能会有人说,随机缩放,普通的数据增强也可以做,但作者考虑到很多人可能只有一个GPU,因此Mosaic增强训练时,可以直接计算4张图片的数据,使得Mini-batch大小并不需要很大,一个GPU就可以达到比较好的效果。

(2)BackBone

CSPDarknet53

CSPDarknet53是在Yolov3主干网络Darknet53的基础上,借鉴2019年CSPNet的经验,产生的Backbone结构,其中包含了5个CSP模块。

每个CSP模块前面的卷积核的大小都是3 * 3,stride=2,因此可以起到下采样的作用。因为Backbone有5个CSP模块,输入图像是608 * 608,所以特征图变化的规律是:608->304->152->76->38->19。

Yolov4在主干网络Backbone采用CSPDarknet53网络结构,主要有三个方面的优点:

  • 优点一:增强CNN的学习能力,使得在轻量化的同时保持准确性。
  • 优点二:降低计算瓶颈
  • 优点三:降低内存成本

Mish激活函数

和Leaky_relu激活函数的图形对比如下:

Yolov4的Backbone中都使用了Mish激活函数,而后面的网络则还是使用leaky_relu函数

Dropblock

Dropblock,其实和常见网络中的Dropout功能类似,也是缓解过拟合的一种正则化方式。

中间Dropout的方式会随机的删减丢弃一些信息,但Dropblock的研究者认为,卷积层对于这种随机丢弃并不敏感,因为卷积层通常是三层连用:卷积+激活+池化层,池化层本身就是对相邻单元起作用。即使随机丢弃,卷积层仍然可以从相邻的激活单元学习到相同的信息。

因此,在全连接层上效果很好的Dropout在卷积层上效果并不好。所以右图Dropblock的研究者则干脆整个局部区域进行删减丢弃

Dropblock的研究者与Cutout进行对比验证时,发现有几个特点:

  • 优点一:Dropblock的效果优于Cutout
  • 优点二:Cutout只能作用于输入层,而Dropblock则是将Cutout应用到网络中的每一个特征图上
  • 优点三:Dropblock可以定制各种组合,在训练的不同阶段可以修改删减的概率,从空间层面和时间层面,和Cutout相比都有更精细的改进。

(3)Neck

在目标检测领域,为了更好的提取融合特征,通常在Backbone和输出层,会插入一些层,这个部分称为Neck。相当于目标检测网络的颈部,也是非常关键的。

Yolov4的Neck结构主要采用了SPP模块、FPN+PAN的方式。

SPP模块

在Yolov4中,SPP模块仍然是在Backbone主干网络之后

作者在SPP模块中,使用k={1x1 , 5x5 , 9x9 , 13x13}的最大池化的方式,再将不同尺度的特征图进行Concat操作。(这里最大池化采用padding操作,移动的步长为1,比如13×13的输入特征图,使用5×5大小的池化核池化,padding=2,因此池化后的特征图仍然是13×13大小)

采用SPP模块的方式,比单纯的使用k*k最大池化的方式,更有效的增加主干特征的接收范围,显著的分离了最重要的上下文特征。

FPN+PAN

Yolov3中具有FPN结构,Yolov4中Neck这部分除了使用FPN外,还在此基础上使用了PAN结构:

Yolov4在FPN层的后面还添加了一个自底向上的特征金字塔。其中包含两个PAN结构。这样结合操作,FPN层自顶向下传达强语义特征,而特征金字塔则自底向上传达强定位特征,两两联手,从不同的主干层对不同的检测层进行参数聚合。

需要注意的是,Yolov4中:

  • 第一个Yolo层是最大的特征图76*76,mask=0,1,2,对应最小的anchor box。
  • 第二个Yolo层是中等的特征图38*38,mask=3,4,5,对应中等的anchor box。
  • 第三个Yolo层是最小的特征图19*19,mask=6,7,8,对应最大的anchor box。

(4)Prediction

CIOU_loss

CIOU_Loss和DIOU_Loss前面的公式都是一样的,不过在此基础上还增加了一个影响因子,将预测框和目标框的长宽比都考虑了进去。

其中v是衡量长宽比一致性的参数,我们也可以定义为:

综合的看下各个Loss函数的不同点:

  • IOU_Loss:主要考虑检测框和目标框重叠面积。
  • GIOU_Loss:在IOU的基础上,解决边界框不重合时的问题。
  • DIOU_Loss:在IOU和GIOU的基础上,考虑边界框中心点距离的信息。
  • CIOU_Loss:在DIOU的基础上,考虑边界框宽高比的尺度信息。

DIOU_nms

Nms主要用于预测框的筛选,常用的目标检测算法中,一般采用普通的nms的方式,Yolov4则采用DIOU_nms

在重叠目标的检测中,DIOU_nms的效果优于传统的nms。

5.YOLO_V5

参考文章:深入浅出Yolo系列之Yolov5核心基础知识完整讲解

Yolov5官方代码中,给出的目标检测网络中一共有4个版本,分别是Yolov5s、Yolov5m、Yolov5l、Yolov5x四个模型。

Yolov5s整体的网络结构图如下:

Yolov5s网络是Yolov5系列中深度最小,特征图的宽度最小的网络。后面的3种都是在此基础上不断加深,不断加宽。

(1)输入端

输入端运用了Mosaic数据增强、自适应锚框计算、自适应图片缩放三种方法,Mosaic数据增强上文介绍过不再赘述。

自适应锚框计算

在Yolo算法中,针对不同的数据集,都会有初始设定长宽的锚框。

在Yolov3、Yolov4中,训练不同的数据集时,计算初始锚框的值是通过单独的程序运行的。

但Yolov5中将此功能嵌入到代码中,每次训练时,自适应的计算不同训练集中的最佳锚框值。当然,如果觉得计算的锚框效果不是很好,也可以在代码中将自动计算锚框功能关闭。

自适应图片缩放

在常用的目标检测算法中,不同的图片长宽都不相同,因此常用的方式是将原始图片统一缩放到一个标准尺寸,再送入检测网络中。

比如Yolo算法中常用416x416,608x608等尺寸,比如对下面800x600的图像进行缩放。

而在Yolov5的代码中datasets.py的letterbox函数中进行了修改,对原始图像自适应的添加最少的黑边

计算方法如下:

  1. 计算缩放比例:原始缩放尺寸除以原始图像的尺寸后,可以得到两个缩放系数,选择小的缩放系数。
  2. 计算缩放后的尺寸:原始图片的长宽都乘以最小的缩放系数。
  3. 计算黑边填充数值:类似下图,使高能够被32整除

(2)Backbone

Backbone中利用了Focus结构,CSP结构

Focus结构


以Yolov5s的结构为例,原始6086083的图像输入Focus结构,采用切片操作,先变成30430412的特征图,再经过一次32个卷积核的卷积操作,最终变成30430432的特征图。

CSP结构

Yolov5中设计了两种CSP结构,以Yolov5s网络为例,CSP1_X结构应用于Backbone主干网络,另一种CSP2_X结构则应用于Neck中。

CSP结构上文已介绍

(3)Neck:FPN+PAN

Yolov5现在的Neck和Yolov4中一样,都采用FPN+PAN的结构,Yolov5和Yolov4的不同点在于,

Yolov4的Neck结构中,采用的都是普通的卷积操作。而Yolov5的Neck结构中,采用借鉴CSPnet设计的CSP2结构,加强网络特征融合的能力。

(4)输出端

Yolov5中采用其中的GIOU_Loss做Bounding box的损失函数。

Yolov5中采用加权nms的方式。

(5)四种网络对比

深度对比


需要注意的是,四种网络结构中每个CSP结构的深度都是不同的。

宽度对比


四种yolov5结构在不同阶段的卷积核的数量都是不一样的,因此也直接影响卷积后特征图的第三维度,即厚度

三、RetinaNet

参考代码:pytorch-retinanet
参考文章:RetinaNet 论文和代码详解

RetinaNet针对现有单阶段法目标检测模型中前景(positive)和背景(negatives)类别的不平衡问题,提出了一种叫做Focal Loss的损失函数,用来降低大量easy negatives在标准交叉熵中所占权重(提高hard negatives所占权重)。

下图为RetinaNet的结构图,我们可以看出,RetinaNet的特征提取网络选择了残差网络ResNet,特征融合这块选择了FPN(特征金字塔网络),以特征金字塔不同的尺寸特征图作为输入,搭建三个用于分类和框回归的子网络。

RetinaNet最主要的创新点在于其使用Focal Loss解决前景和背景样本不平衡的问题,Focal Loss损失函数输入为网络的分类预测,框回归预测,anchor信息(用于对框回归预测进行转换)和真实标签annotations。

我们要解决样本不平衡的问题,我们使用一个平衡因子α ,其范围是0到1,对于类别1乘以 α ,而对于类别-1乘以 1-α,则损失函数变为

虽然 α 平衡了正负样本的数量,但实际上,目标检测中大量的候选目标都是易分样本。这些样本的损失很低,但是由于数量极不平衡,易分样本的数量相对来讲太多,最终主导了总的损失。因此,Focal Loss认为易分样本(即,置信度高的样本)对模型的提升效果非常小,模型应该主要关注与那些难分样本 。一个简单的想法就是将高置信度样本的损失降低一些, 也即是下面的公式:

将 γ 定义为2时效果最好!

四、Anchor-free模型

参考文章:
目标检测:Anchor-Free时代
目标检测Anchor-free分支:基于关键点的目标检测

目标检测综述——单阶段检测器相关推荐

  1. 基于深度学习的目标检测综述(单阶段、多阶段、FPN变体、旋转目标检测等)

    随着深度学习的发展,基于深度学习的目标检测方法因其优异的性能已经得到广泛的使用.目前经典的目标检测方法主要包括单阶段(YOLO.SSD.RetinaNet,还有基于关键点的检测方法等)和多阶段方法(F ...

  2. 目标检测综述——两阶段检测器

    文章目录 1.RCNN 2.SPPNet 3.Fast RCNN 4.Faster RCNN 5.Feature Pyramid Networks(FPN) 6.其他的变种 参考综述文章: 论文笔记- ...

  3. 【目标检测】单阶段算法--YOLOv4详解

    论文题目 :<YOLOv4: Optimal Speed and Accuracy of Object Detection> 论文地址:https://arxiv.org/pdf/2004 ...

  4. 【目标检测】单阶段算法--YOLOv2详解

    论文题目:<YOLO9000: Better, Faster, Stronger> 论文地址:https://arxiv.org/pdf/1612.08242.pdf 一文读懂YOLOv1 ...

  5. 【目标检测】单阶段算法--YOLOv1详解

    论文题目:<You Only Look Once:Unified, Real-Time Object Detection> 论文地址:https://arxiv.org/pdf/1506. ...

  6. 【目标检测】单阶段算法--YOLOv3详解

    论文题目:<YOLOv3: An Incremental Improvement > 论文地址:https://pjreddie.com/media/files/papers/YOLOv3 ...

  7. 2022目标检测综述

    目录 0.引言 1.背景 1.1.问题描述 1.2.目标检测的核心问题 1.3.目标检测中的关键挑战 2.关于损失函数 3.关于IOU 4.数据集以及评价指标 4.1.评价指标 4.2.数据集 5.目 ...

  8. 工业应用中基于三维模型的6D目标检测综述

    文章名称:A Survey of 6D Object Detection Based on 3D Models for Industrial Applications,J. Imaging | Fre ...

  9. Deep Learning for Generic Object Detection: A Survey -- 目标检测综述总结

      最近,中国国防科技大学.芬兰奥卢大学.澳大利亚悉尼大学.中国香港中文大学和加拿大滑铁卢大学等人推出一篇最新目标检测综述,详细阐述了当前目标检测最新成就和关键技术.文章最后总结了未来8个比较有前景的 ...

最新文章

  1. 解决git提交时报“there are no staged files”
  2. postgres中文文档
  3. 图解Oracle 11g physical standby Rolling Upgrade物理备库滚动升级特性
  4. CFA将人工智能列入考试内容折射啥
  5. 在 k8s 以外的分布式环境中使用 Dapr
  6. java的配置文件后缀,Java - 敏感配置文件位置
  7. 专题导读:大数据整理
  8. Bootstrap按钮的外观
  9. 创建list集合_Java 9新增的不可变集合
  10. 计算机多功能发展,随着电子计算机向微型化和多功能化发展,它的体积大大缩小,成本大大降低。性能明 - 问答库...
  11. xstatic图形连接linux,RHCE认证考试模拟题深入讲解:linux配置静态IP地址为:192.168.0.X!...
  12. 使用Python写登录京东商城购物,加入购物车的脚本
  13. word导出手机端乱码或者打不开解决办法
  14. Qt实现读取BIN文件
  15. Laravel 全汉字首字母分类排序
  16. Java中ch.ethz.ssh2.Connection对象的使用
  17. 黑鲨1 救黑砖 9008救砖
  18. 期权、期货及其他衍生产品 第四章读书笔记 利率
  19. 神经网络epoch和batch的粗浅理解
  20. SQL server 数据库关系图及数据完整性设计

热门文章

  1. GAN诞生记:最火的AI模型,来自一群博士的酒后争吵
  2. 例14:计算某日是该年的第几天
  3. NSCharacterSet 简单用法
  4. I.MX6 U-boot Kernel backlight setting
  5. 2012年云计算领域将发生6件大事
  6. Lucene.Net 初学笔记 - 索引
  7. 华为机试HJ24:合唱队
  8. matlab 离散傅里叶变换_时域上的采样VS频域上的周期延拓。用离散傅里叶变换直观地演示...
  9. 如果项目上线在上线期间出现BUG改怎么办?
  10. 修改时间日期 linux,查看和修改 Linux 的时间和日期