文章目录
- 前言
- 一、YOLOv2的网络结构和流程
- 二、YOLOv2的创新点
- 预处理
- 网络结构
- 训练
- 总结
前言
根据前文目标检测-One Stage-YOLOv1可以看出YOLOv1的主要缺点是：
- 和Fast-CNN相比，速度快，但精度下降。（边框回归不加限制）
YOLOv2提出了一些改进策略，如anchor-based等
提示：以下是本篇文章正文内容，下面内容可供参考
一、YOLOv2的网络结构和流程
- 将影像输入卷积网络（DarkNet-19+残差连接）得到13 × 13特征图
- 引入anchor机制，与SSD不同的是，每个特征点对应5个anchor，且anchor的大小是由VOC 和 COCO数据集聚类得到的
ps：由于变为anchor-based算法，预测框由YOLOv1的98个变为845（13 × 13 × 5）个，mAP由69.5略微降到69.2，召回率却由81大大提升至88
- 将上一步得到的anchor输入分类和边框回归器
- 使用非极大值抑制NMS去除冗余窗口
下图可以比较清晰的看出YOLOv2的主要结构
其中，DarkNet-19的结构如下：
更详细的参数如下：
ps：上图中可以看出，残差连接时要保证两个特征图w，h的一致（从通道上进行拼接），这里通过PassThrough层将之前层的特征图进行了w，h的缩放，以和更深层特征图保持一致，PassThrough层的基本原理如下图：
之所以不用池化下采样，是想通过PassThrough保留featureMap的更多细节
二、YOLOv2的创新点
预处理
- 使用了标准的数据增强方法：随机裁剪、旋转（random crops, rotations）；色调、饱和度（hue, saturation）；曝光偏移（exposure shifts）
网络结构
- backbone：改为Darknet-19，Darknet-19 的性能基本与 Resnet34 差不多，使得网络更轻量更快
- 引入了BN（Batch normalization），其优点如下：
- 加快收敛；
- 改善梯度，远离饱和区；
- 允许大的学习率；
- 对初始化不敏感；
- 相当于正则化，使得有BN层的输入都有相近的分布；
- 有了BN之后，就可以不用dropout了，或者说不能像原来一样用dropout了，这会导致训练和测试的方差偏移。
- 加入了anchor机制
- 细粒度特征（Fine-Grained Features）：将最后一个最大池化层前的特征图经过Pass Through与后面的卷积特征图进行合并，Pass Through就是四分后再concat
训练
- 高分辨率的预训练，采用了448的输入进行微调，以便网络更好地处理更高分辨率的输入
- 多尺度训练：Yolov2每10个batches就会随机换一下输入的尺度（{320, 352, …, 608}），使得模型泛化于不同尺度的输入，这得益于adaptive pooling层。高分辨率的输入速度慢，但是对小目标的检测效果要好很多，低分辨率的输入速度快
- loss略微改动：（真阳样本的定位误差、confidence误差、分类误差）、预测框和anchor定位误差、负样本置信度误差，且5个子loss均有一个权重超参数
- 预测输入从448变为416，目的是让得到的feature map的size是一个奇数。这样的好处是，许多图片的中心点都是某个物体的中心，奇数保证中间是一个格子，而不是偶数那样四个格子抢占中心点
- 尝试了一种分类和检测的联合训练策略，类别数据集用于分类训练，检测数据集用于边框回归和分类，同时为了统一COCO数据集和ImageNet数据集类别，提出了一种层级分类方法
总结
通过网络结构和改进和一些tricks，YOLOv2精度大大提升，同时通过多尺度训练，使得相同的YOLOv2模型可以在不同的大小下运行，从而轻松实现速度和精度之间的折衷。
- 在67 FPS的速度下，YOLOv2在VOC 2007上达到76.8mAP。
- 在40 FPS的速度下，YOLOv2在VOC 2007上达到78.6mAP，性能优于当时的SOTA，如SSD和Faster RCNN ResNet，同时运行速度更快。
