【You only look once】目标检测YOLO系列算法改进史

发布于:2024-08-26 ⋅ 阅读:(70) ⋅ 点赞:(0)


YOLOv1

  • 核心思想:
    将目标检测任务视为一个回归问题,直接在整幅图像上进行边界框的预测和类别分类,一次性输出目标的位置和类别信息。
  • 网络结构: 输入图像被 resize 到 448x448,经过一系列卷积层和全连接层处理。最后一个全连接层输出一个 7x7x30 的张量,其中 7x7 表示将图像划分为 7x7 的网格,每个网格对应 30 维的预测信息,包括 2 个边界框的位置(4 个值)、置信度(2 个值)和 20 类对象分类的概率。
  • 优点: 检测速度快,能够在一张图像上同时预测多个目标的位置和类别;对背景的误检率相对较低,因为是基于整幅图像进行检测,能利用上下文信息。
  • 缺点: 定位精度不够高,尤其是对于小目标的检测效果较差;每个网格只能预测一组类别概率,对于多个目标位于同一网格的情况处理能力有限;边界框的预测不够准确,召回率相对较低。

YOLOv2

  • 改进之处:
    1. 输入尺度: 支持更高分辨率的输入,提高了对小目标的检测能力。
    2. 网络层改进:
      • 在卷积和激活函数之间添加批量归一化(Batch Normalization),加快训练速度并提高准确率,减少了对 Dropout 层的依赖。
      • 借鉴 ResNet 的思想,进行特征图的拼接操作,融合不同层次的特征信息,增强了模型对多尺度特征的表达能力。
    3. 输出结果处理:
      • 引入先验框(Anchor Boxes)的概念,每个网格预先定义多个不同尺寸和比例的先验框,提高了对不同形状目标的适应性,虽然准确率略有下降,但召回率提升明显。
      • 对边界框的预测值进行编码(Encode)处理,将预测值的范围控制在一定范围内,使模型训练更加稳定。
  • 性能提升: 在保持较快检测速度的同时,一定程度上提高了检测精度和召回率,尤其是对小目标的检测效果有所改善。

YOLOv3

  • 改进之处:
    1. 多尺度预测: 在三个不同尺度上进行边界框的预测,利用类似特征金字塔网络的思想,融合不同层次的特征图,提高了对不同大小目标的检测能力,尤其是对小目标的检测效果进一步提升。
    2. 特征提取器: 采用新的 Darknet-53 作为特征提取器,它是 YOLOv2 中的 Darknet-19 与残差链接混合的方法改进而来,具有更深的网络层次(53 层),性能优于 ResNet-101,速度更快,能够提取更丰富、更具代表性的特征。
    3. 边界框预测: 沿用 YOLOv2 的维度聚类方式生成先验框,并为每个边界框预测一个物体得分,使用逻辑回归进行预测,同时采用单独的逻辑分类器对每个类别进行预测,不使用 Softmax 分类器,以适应一些标签不完全互斥的数据集。
  • 性能特点: 检测精度较高,速度也相对较快,在不同尺度的目标检测上表现较为均衡,能够处理较为复杂的检测场景。

YOLOv4

  • 改进之处:
    1. 骨干网络: 采用了更强大的 CSPDarknet53 作为骨干网络,结合了跨阶段局部网络(CSPNet)的思想,增强了网络的特征提取能力和学习能力,提高了模型的准确性和效率。
    2. 数据增强: 使用了多种数据增强技术,如 Mosaic 数据增强,通过将多张图片拼接在一起进行训练,丰富了训练数据的多样性,提高了模型的泛化能力和鲁棒性。
    3. 激活函数: 使用 Mish 激活函数代替传统的 Leaky ReLU 激活函数,Mish 函数在负值区域有更平滑的过渡,有助于提高模型的收敛速度和准确性。
    4. 优化算法: 在训练过程中采用了一些优化技巧,如学习率余弦退火衰减、标签平滑等,进一步提高了模型的训练效果和稳定性。
    5. 检测头: 对检测头进行了改进,采用了更高效的结构,提高了检测的速度和精度。
  • 性能优势: 在保持较高检测速度的前提下,显著提高了检测精度,尤其是在复杂场景下对各种目标的检测能力较强,具有较好的泛化性能和鲁棒性。

YOLOv5

  • 改进之处:
    1. 灵活的模型架构: YOLOv5 提供了多种不同规模的模型,如 YOLOv5s、YOLOv5m、YOLOv5l、YOLOv5x 等,以适应不同的应用需求和计算资源。用户可以根据实际情况选择合适的模型,在速度和精度之间进行权衡。
    2. 更高效的训练策略: 采用了自适应锚框计算、自适应图像缩放等技术,提高了训练效率和模型的适应性。同时,在数据增强方面也进行了优化,使用了多种数据增强方法的组合,进一步提升了模型的性能。
    3. 优化的网络结构: 对网络的各个组件进行了优化,如改进了骨干网络的结构,使其在特征提取方面更加高效;优化了颈部网络(Neck)和头部网络(Head)的设计,提高了特征融合和目标检测的效果。
  • 性能特点: 具有较高的检测精度和较快的检测速度,训练和部署相对简单方便,能够在各种硬件平台上高效运行,广泛应用于实际场景中的目标检测任务。

YOLOv6

  • 改进之处:
    1. 骨干网络: 使用了更高效的 EfficientRep 作为骨干网络,它是基于 RevGG 结构改进而来,对硬件更加友好,能够在 GPU 设备上实现更高效的计算,提高了模型的推理速度。
    2. 颈部网络: 基于 Rep 和 PAN 搭建了 Rep-PAN 颈部网络,通过引入 Rep 结构,降低了硬件上的延时,提高了特征融合的效率。
    3. 检测头: 与 YOLOX 一样对检测头进行解耦,将边框回归与类别分类分开,提高了模型的性能和收敛速度。同时,重新设计了一个更高效的解耦头结构,在不影响精度的情况下降低了延时,实现了速度与精度的平衡。
    4. 标签分配: 沿用了 YOLOX 的 SimOTA 标签分配方法,能够为不同目标设定不同的正样本数量,解决了正负样本不均衡的问题,提高了模型的训练效果。
    5. 边框回归损失: 引入了新的边框回归损失 SIoU,通过考虑框与框之间的向量角度,重新定义了距离损失,有效降低了回归的自由度,加速了网络的收敛,进一步提升了回归的准确性。
  • 性能表现: 在保持较高检测精度的同时,大幅提高了推理速度,尤其在工业界的实际应用中具有很大优势,能够满足对实时性要求较高的场景需求。

YOLOv7

  • 改进之处:
    1. 网络结构设计:
      • 使用了 ELAN 和 MP 结构等新颖的模块。ELAN 结构通过控制最短和最长的梯度路径,使更深的网络能够更有效地学习和收敛,并且基于 ELAN 设计的 E-ELAN 进一步增强了网络的学习能力。MP 结构则同时采用了最大池化和步长为 2 的卷积,在不增加太多计算量的情况下,丰富了特征提取的方式。
      • 采用了重参数化结构(Rep structure),将多个分支的参数进行整合,在训练时具有多分支结构的优势,而在推理时可以等效为单路模型,提高了推理速度。
    2. 激活函数: 使用 SiLU 激活函数,相比传统的激活函数,SiLU 在负值区域有更平滑的过渡,有助于提高模型的收敛速度和准确性。
    3. 损失函数和匹配策略: 在损失函数方面,整体和 YOLOv5 保持一致,分为坐标损失、目标置信度损失和分类损失三部分,并采用了 CIoU 损失作为坐标损失函数。在匹配策略上,参考了 YOLOv5 和 YOLOv6 使用的 SimOTA 方法,并进行了一些改进和优化,如在训练前通过 k-means 聚类算法获得先验的 anchor 框,根据 gt 框的中心位置扩充正样本等,提高了模型的训练效果和对目标的检测能力。
  • 性能特点: 在速度和精度方面都有较好的表现,尤其是在处理复杂场景和小目标检测方面具有一定优势,能够在不同的硬件平台上实现高效的目标检测。

YOLOv8

  • 改进之处:
    1. 统一的模型架构: YOLOv8 采用了统一的模型架构,包括骨干网络、颈部网络和检测头,使得不同规模的模型(如 YOLOv8n、YOLOv8s、YOLOv8m、YOLOv8l、YOLOv8x)之间具有更好的一致性和可扩展性。
    2. 更强大的骨干网络: 使用了新的骨干网络结构,如 C2f、C3 等模块,能够更有效地提取图像特征,提高模型的检测精度和速度。
    3. 高效的检测头: 检测头部分进行了优化,采用了更先进的卷积结构和注意力机制,提高了对目标的定位和分类准确性。
    4. 先进的训练策略: 采用了诸如标签平滑、IoU-aware loss 等训练技巧,进一步提升了模型的性能和泛化能力。
    5. 多任务学习支持: 除了目标检测任务,YOLOv8 还支持实例分割等多任务学习,能够满足更多不同的应用需求。
  • 性能优势: 在保持高速度的同时,显著提高了检测精度,尤其是在小目标检测和密集目标检测场景下表现出色。模型具有较高的灵活性和可定制性,能够适应不同的应用场景和硬件环境。

待更新

随着技术的不断发展,它们可能在以下方面进行改进和创新:

  1. 模型结构优化: 进一步改进骨干网络、颈部网络和头部网络的结构,提高特征提取和融合能力,以提高检测精度。
  2. 检测速度提升: 通过优化算法和硬件加速等方式,进一步提高检测速度,满足实时性要求更高的应用场景。
  3. 小目标检测: 持续改进对小目标的检测效果,提高在复杂场景下对小目标的识别能力。
  4. 多模态融合: 结合其他传感器数据或模态信息,如红外图像、深度信息等,提高检测的准确性和鲁棒性。
  5. 模型压缩和轻量化: 开发更高效的模型压缩技术,使模型在保持高性能的同时,能够在资源受限的设备上运行。
  6. 自适应性和泛化能力: 提高模型对不同数据集和场景的自适应性和泛化能力,减少对大量标注数据的依赖。

提示:这些版本之间的关联在于它们都是在不断改进和优化目标检测算法,以满足不同应用场景的需求。每个版本都继承了前一版本的优点,并针对前一版本的不足进行改进和创新。同时,它们也为后续版本的发展提供了经验和基础。


网站公告

今日签到

点亮在社区的每一天
去签到