术语表

数据扩充

利用数据扩充提高模型准确性!学习扩展数据集、防止过度拟合和提高计算机视觉性能的技术。

使用Ultralytics HUB 对YOLO 模型进行简单培训

了解更多

数据扩增是机器学习中的一种技术,通过创建现有数据的修改版本,人为地增加训练数据集的规模。这种技术有助于提高机器学习模型的性能和泛化能力,尤其是在原始数据集有限的情况下。通过让模型接触到更广泛的数据点,数据扩增有助于防止过拟合,即模型过于专注于训练数据,而在未见数据上表现不佳。

数据扩充的相关性和重要性

计算机视觉任务中,收集和标记大型数据集既昂贵又耗时,因此数据扩增显得尤为重要。例如,在物体检测中,在有限的数据集上训练出来的模型可能难以识别不同光照条件、方向或比例下的物体。通过应用旋转、翻转、裁剪和颜色调整等增强功能,模型可以适应更广泛的场景,从而在实际应用中更加稳健和准确。数据增强还可用于解决类不平衡问题,即数据集中某些类的样本明显少于其他类。通过有选择地扩充代表性不足的类别的数据,模型可以学习到更均衡的数据表示。了解有关计算机视觉项目数据收集和标注的更多信息。

常见的数据扩充技术

计算机视觉中常用的数据增强技术包括

  • 几何变换:这涉及到改变图像的空间属性,例如:
    • 旋转将图像旋转一定角度。
    • 翻转:水平或垂直翻转图像。
    • 缩放:放大或缩小图像
    • 裁剪:提取图像的一部分。
    • 平移水平或垂直移动图像。
  • 色彩空间转换:这些涉及修改图像的颜色属性,例如
    • 亮度调整:增加或减少图像的整体亮度。
    • 对比度调整:增大或减小图像最暗和最亮部分之间的差异。
    • 饱和度调整:增加或减少图像中的色彩强度。
    • 色调调整:沿色谱移动图像中的颜色。
  • 噪声注入:向图像中添加随机噪音,这可以帮助模型更好地适应图像质量的变化。
  • 内核滤镜对图像应用滤镜,如模糊或锐化,以模拟不同的图像捕捉条件。

现实世界应用中的数据增强

数据增强被广泛应用于现实世界中的各种人工智能/人工智能应用中。下面是两个具体的例子:

  1. 医学图像分析:在医学影像领域,由于隐私问题和对专家注释的需求,获取大型标记数据集具有挑战性。数据增强技术可用于增强医学图像数据集的规模和多样性,从而提高肿瘤检测、病变分割和疾病分类等任务模型的性能。例如,在脑肿瘤检测数据集中,对现有的 MRI 或 CT 扫描应用旋转、翻转和小幅平移可以帮助模型学会识别肿瘤,而不管它们在大脑中的方向或位置如何。探索医疗保健领域的人工智能,了解更多信息。
  2. 自动驾驶:自动驾驶汽车主要依靠计算机视觉模型来感知环境,包括检测行人、车辆和交通标志。数据增强在训练这些模型以适应各种实际条件方面发挥着至关重要的作用。例如,通过在训练图像中添加不同的光照条件、天气条件(如雨、雪、雾)和遮挡物,模型可以学会在具有挑战性的场景下准确检测物体,从而提高自动驾驶系统的安全性和可靠性。了解有关自动驾驶汽车中的人工智能的更多信息。

数据扩充及相关概念

数据增强与机器学习中的其他重要概念密切相关:

  • 过度拟合:数据扩增是缓解过拟合的有力技术,即模型在训练数据上表现良好,但在未见数据上表现不佳。通过增加训练集的多样性,扩增有助于模型更好地泛化到新的、未见过的示例中。这与欠拟合形成鲜明对比,欠拟合是指模型过于简单,无法捕捉数据中的潜在模式。
  • 正则化:数据扩增可视为正则化的一种形式,因为它引入了一些约束条件,防止模型变得过于复杂,与训练数据拟合得过于紧密。其他正则化技术包括 L1 和 L2 正则化,它们根据权重的大小对模型的损失函数进行惩罚。
  • 迁移学习:数据扩增与迁移学习相结合会特别有效,在迁移学习中,在大型数据集(如 ImageNet)上预先训练好的模型会在较小的、针对特定任务的数据集上进行微调。增强较小的数据集有助于模型更好地适应新任务,并防止对有限数据的过度拟合。了解有关预处理注释数据的更多信息。

总之,数据增强是提高机器学习模型性能和鲁棒性的关键技术,尤其是在计算机视觉领域。通过使用现有数据的修改版本人为地扩展训练数据集,数据扩增有助于模型更好地泛化到未见数据,防止过拟合,并解决类不平衡等问题。Ultralytics YOLO 模型利用数据扩增来提高它们在各种计算机视觉任务中的性能。Albumentations与Ultralytics 的集成提供了一种强大而灵活的方式,可将各种增强功能应用于您的数据集。访问我们的集成页面,了解有关Ultralytics 如何与其他工具和平台协同工作的更多信息。

阅读全部