Ultralytics YOLO27: 加入候补名单。
获取 YOLO 许可
指南

探索 2025 年最佳计算机视觉数据集

探索 ImageNet、COCO、DOTA-v2.0、Roboflow 100 和 Open Images V7 等顶级计算机视觉数据集,并了解如何为项目选择合适的数据集。

Abirami VinaAbirami Vina8 分钟阅读
用于模型训练的计算机视觉数据集

你知道吗?数据在你日常生活中几乎无处不在。观看视频、拍照或查看 Google 地图,都会为超过 750 亿台联网设备持续捕获的信息流添砖加瓦。这些数据构成了人工智能 (AI) 的基础。事实上,Ultralytics YOLO11 等先进计算机视觉模型依靠视觉数据来识别模式、解读图像,并理解我们周围的世界。

有趣的是,数据的价值并不只取决于数量,数据的组织和准备情况更为重要。如果数据集杂乱或不完整,就可能导致错误。而干净且多样化的数据集能帮助计算机视觉模型提升性能,无论是识别人群中的物体,还是分析复杂图像。高质量数据集至关重要。

本文将介绍 2025 年最佳计算机视觉数据集,并探讨这些数据集如何帮助构建更准确、更高效的计算机视觉模型。

什么是计算机视觉数据集?#

计算机视觉数据集是由图像或视频组成的集合,可帮助计算机视觉系统学习理解和识别视觉信息。这些数据集带有标签或注释,可帮助模型识别数据中的物体、人物、场景和模式。

这些数据集可用于训练计算机视觉模型,帮助模型改进人脸识别、目标检测或场景分析等任务。数据集越优质——组织越有序、多样性越丰富、准确性越高——视觉 AI 模型的表现就越好,最终让日常生活中的技术变得更智能、更实用。

如何构建计算机视觉数据集#

构建计算机视觉数据集就像准备学习笔记,教别人观察和理解这个世界。一切都从收集与你正在开发的特定应用相匹配的图像和视频开始。

理想的数据集应包含感兴趣物体的多种示例,拍摄角度各异、光照条件不同,并涵盖多种背景和环境。丰富的多样性可确保计算机视觉模型准确学习识别模式,并在现实场景中可靠运行。

构建理想视觉数据集的示意图

图 1. 构建理想的视觉数据集。图片由作者提供。

收集相关图像和视频后,下一步是进行数据标注。这个过程包括为数据添加标签、注释或描述,让 AI 能够理解每张图像或每段视频包含的内容。

标签可以包含物体名称、位置、边界或其他相关细节,帮助模型准确学习识别和解读视觉信息。数据标注将简单的图像集合转化为结构化数据集,可用于训练计算机视觉模型。

模型训练需要高质量数据#

你可能想知道什么样的数据集才算高质量。影响因素有很多,例如标注的准确性、多样性和一致性。比如,多个标注人员为目标检测数据集标注猫耳时,一个人可能把猫耳标为头部的一部分,另一个人却把猫耳单独标出。这种不一致可能会让模型困惑,并影响模型正确学习的能力。

下面快速了解理想计算机视觉数据集的特点:

  • 标签清晰:每张图像都经过准确标注,标签一致且精确。
  • 数据多样:数据集包含不同的物体、背景、光照条件和拍摄角度,帮助模型在各种情况下都能良好运行。
  • 高分辨率图像:清晰、细节丰富的图像更有利于模型学习和识别特征。

Ultralytics 支持多种数据集#

Ultralytics YOLO 模型(例如 YOLO11)专为使用特定的 YOLO 文件格式数据集而构建。虽然你可以轻松将自己的数据转换为这种格式,但我们也提供了更简单的选项,方便你立即开始试用。

Ultralytics Python 包支持多种计算机视觉数据集,让你无需额外设置即可开展项目,使用目标检测、实例分割或姿态估计等任务。

用户只需在训练函数的参数中指定数据集名称,就能轻松使用 COCO、DOTA(v1.0 和 v1.5)、Open Images V7 和 ImageNet 等现成数据集。指定后,系统会自动下载并预配置数据集,让你专注于构建和改进模型。

2025 年五大计算机视觉数据集#

视觉 AI 的进步离不开多样化的大规模数据集,这些数据集推动创新并促成突破。下面来看看 Ultralytics 支持的一些重要数据集,以及它们如何影响计算机视觉模型。

ImageNet 数据集#

ImageNet由 Fei-Fei Li 及其普林斯顿大学团队于 2007 年创建,并于 2009 年推出,是一个包含超过 1,400 万张标注图像的大型数据集。它广泛用于训练系统识别和分类不同物体。其结构化设计对于教会模型准确分类图像尤其有用。尽管文档齐全,它主要侧重于图像分类,且缺少物体检测等任务所需的详细标注。

下面来看看 ImageNet 的一些主要优势:

  • 多样性: ImageNet 的图像涵盖 20,000 多个类别,提供了海量且多样的数据集,有助于提升模型训练效果和泛化能力。
  • 结构化组织: 图像按照 WordNet 层级结构经过精心分类,便于高效检索数据和系统地训练模型。
  • 全面的文档:经过广泛研究和多年积累,ImageNet 的文档对初学者和专家都很有帮助,能为计算机视觉项目提供宝贵的见解和指导。

不过,和任何数据集一样,它也有局限性。以下是需要考虑的一些挑战:

  • 计算资源需求: 数据集规模庞大,可能给计算资源有限的小型团队带来挑战。
  • 缺少时间数据: 由于数据集仅包含静态图像,因此可能无法满足需要视频或时序数据的应用需求。
  • 图像过时: 数据集中的一些图像年代较久,可能无法反映当前的物体、风格或环境,从而降低其对现代应用的相关性。

DOTA-v2.0 数据集#

DOTA-v2.0 数据集是专为旋转边界框(OBB)物体检测创建的大型航拍图像集合,其中 DOTA 是“航拍图像物体检测数据集”的缩写。在 OBB 检测中,旋转边界框用于更准确地贴合图像中物体的实际朝向。这种方法尤其适用于航拍图像,因为其中的物体通常呈现各种角度,因此能够实现更精确的定位和更好的整体检测效果。

该数据集包含 11,000 多张图像和 18 个物体类别中的 170 多万个旋转边界框。图像尺寸从 800×800 到 20,000×20,000 像素不等,其中包括飞机、船只和建筑物等物体。

DOTA-v2.0 数据集中的示例图像和标注

图 2。 DOTA-v2.0 数据集中的图像和标注示例。图片由作者提供。

凭借详细的标注,DOTA-v2.0 已成为遥感和空中监视项目的热门选择。以下是 DOTA-v2.0 的一些主要特性:

  • 多样的物体类别: 数据集涵盖许多不同类型的物体,例如车辆、港口和储罐,让模型接触各种真实世界中的物体。
  • 高质量标注: 专业标注人员提供了方向精确的边界框,清晰展示物体的形状和朝向。
  • 多尺度图像: 数据集包含不同尺寸的图像,帮助模型学习检测小尺度和大尺度物体。

DOTA-v2 优势众多,但用户也应注意以下一些局限性:

  • 额外的下载步骤: 由于 DOTA 数据集的维护方式,DOTA-v2.0 需要额外的设置步骤。你需要先下载 DOTA-v1.0 图像,再添加额外图像和更新后的 DOTA-v2.0 标注,才能完成数据集的构建。
  • 复杂的标注: 在模型训练期间,处理旋转边界框可能需要额外的工作。
  • 适用范围有限: DOTA-v2 专为航拍图像设计,因此在此领域之外的一般物体检测任务中用处较小。

Roboflow 100 数据集#

Roboflow 100(RF100)数据集由 Roboflow 在 Intel 的支持下创建。它可用于测试和评估物体检测模型的表现。这个基准数据集包含从 90,000 多个公共数据集中挑选出的 100 个不同数据集。它有超过 224,000 张图像和 800 个物体类别,覆盖医疗保健、航拍图像和游戏等领域。

以下是使用 RF100 的一些主要优势:

  • 广泛覆盖多个领域: 数据集涵盖七个领域,例如医学成像、航拍图像和水下探索。
  • 推动模型改进: RF100 中的多样性和特定领域挑战揭示了当前模型的不足,推动研究朝着更具适应性和鲁棒性的物体检测解决方案发展。
  • 统一的图像格式: 所有图像都调整为 640x640 像素。这有助于用户训练模型,而无需调整图像尺寸。

尽管 RF100 有不少优点,但也有一些需要注意的缺点:

  • 任务范围有限: RF100 专为物体检测设计,因此无法支持分割或分类等任务。
  • 侧重基准评测: RF100 主要作为基准测试工具设计,而非用于训练适用于真实世界应用的模型,因此其结果可能无法完全适用于实际部署场景。
  • 标注差异: 由于 RF100 汇集了众包数据集,标注质量和标记规范可能存在不一致,进而影响模型评估和微调。

COCO(上下文中的常见物体)数据集#

COCO 数据集是使用最广泛的计算机视觉数据集之一,提供超过 330,000 张带有详细图像标注的图像。它专为物体检测、分割和图像描述设计,是许多项目的重要资源。包括边界框和分割掩码在内的详细标签,有助于系统学习精确分析图像。

该数据集以灵活著称,适用于从简单到复杂的各种任务。它已成为视觉 AI 领域的标准,经常用于挑战赛和竞赛,以评估模型性能。

它的一些优势包括:

  • 多样且真实的数据: 数据集包含来自真实场景的图像,其中有多个物体、遮挡和多变的光照条件。
  • 广泛的社区和研究应用: COCO 数据集用于大型机器学习竞赛和研究,拥有完善的文档、预训练模型以及活跃的社区支持。
  • 丰富细致的标注:COCO 数据集提供了非常细致的标注,包括目标分割、关键点和图像说明,因此非常适合需要精准视觉理解的项目。

以下是一些同样需要注意的局限因素:

  • 计算资源需求高: 由于规模大且复杂,在 COCO 上训练模型可能需要大量计算资源,这对硬件有限的团队来说颇具挑战。
  • 数据不平衡: 某些物体类别的图像数量明显多于其他类别,这可能导致模型训练出现偏差。
  • 标注结构复杂: 数据集的详细标注虽然很有价值,但对于缺少结构化视觉 AI 数据集使用经验的初学者或小型团队来说,可能难以应对。

Open Images V7 数据集#

Open Images V7是由 Google 整理的超大型开源数据集,包含超过 900 万张图像,并提供 600 个物体类别的标注。它包含多种标注类型,非常适合应对复杂的计算机视觉任务。其规模和丰富程度为训练和测试计算机视觉模型提供了全面的资源。

Open Images V7 数据集中的示例图像

图 3。一窥 Open Images V7 数据集。图片由作者提供。

此外,Open Images V7 数据集在研究领域广受欢迎,因此用户可以找到大量学习资源和示例。不过,数据集规模庞大,下载和处理可能耗时很久,尤其是对小型团队而言。另一个问题是,部分标注可能不一致,需要额外工作来清理数据;而且集成过程并非总是顺畅,因此可能需要进行额外准备。

选择合适的数据集#

选择合适的数据集是确保计算机视觉项目取得成功的重要一环。最佳选择取决于你的具体任务——找到匹配的数据集有助于模型学会所需技能。数据集还应能轻松集成到你的工具中,这样你就能把更多精力放在构建模型上,而不是排查问题。

选择合适数据集的影响因素示意图

图 4。选择合适数据集的影响因素。图片由作者提供。

要点#

高质量数据集是任何计算机视觉模型的基础,能帮助系统学会准确解读图像。多样且标注完善的数据集尤为重要,因为它们能让模型在真实场景中可靠运行,并减少因数据有限或质量低下而造成的错误。

Ultralytics 简化了访问和使用计算机视觉数据集的流程,让你更轻松地找到适合项目的数据。选择合适的数据集是构建高性能模型的关键步骤,有助于取得更精准、更具影响力的结果。

加入我们的社区,并探索我们的 GitHub 代码库,了解更多 AI 相关内容。前往我们的解决方案页面,了解医疗保健领域的计算机视觉和自动驾驶汽车中的 AI等进展。查看我们的许可选项,今天就迈出开始使用计算机视觉的第一步!

作者

Abirami Vina
Abirami Vina

内容创作者 · Ultralytics

Abirami Vina 职业生涯始于计算机视觉工程师,由此在视觉 AI 和机器学习领域打下了坚实基础。她是 Ultralytics 的内容创作者,也是 Scribe of AI 的创始人兼首席撰稿人,为专注 AI 的公司撰写高质量技术内容。她热衷于让 AI 更易理解、更具吸引力,通过自己的作品连接前沿技术与现实影响。

评论

提出问题、分享你的结果,或帮助其他人。

探索解决方案

体育领域的计算机视觉

体育领域的计算机视觉

了解 Ultralytics YOLO 如何通过球员和球追踪、表现分析、教练洞察及运动员康复,推动体育视觉技术的发展。
了解详情
航空影像计算机视觉

航空影像计算机视觉

使用 Ultralytics YOLO 将无人机和航空影像转化为农业、水产养殖、保护工作和地理空间分析的实时洞察。
了解详情
航空航天中的计算机视觉

航空航天中的计算机视觉

使用 Ultralytics YOLO 将计算机视觉应用于航空航天领域,包括飞机检查、跑道 FOD 检测、机场地面作业和复合材料缺陷检测。
了解详情
安防领域的计算机视觉

安防领域的计算机视觉

使用 Ultralytics YOLO 模型守护每个场所。视觉 AI 可用于周界监控、威胁检测、车牌识别和工作场所安全。
了解详情
机器人领域的计算机视觉

机器人领域的计算机视觉

使用 Ultralytics YOLO 模型提升智能机器的能力。机器人领域的视觉 AI 可用于自主导航、感知、目标跟踪和实时控制。
了解详情
物流中的计算机视觉

物流中的计算机视觉

借助 Ultralytics YOLO 模型优化物流流程。视觉 AI 可用于包裹检查、分拣、车辆跟踪和仓库安全实时监控。
了解详情
零售中的计算机视觉

零售中的计算机视觉

借助 Ultralytics YOLO 模型重新构想零售。视觉 AI 可用于库存跟踪、货架监控、排队管理和更深入的客户洞察。
了解详情
医疗领域的计算机视觉

医疗领域的计算机视觉

借助 Ultralytics YOLO 模型构建医疗解决方案。医疗领域的视觉 AI 可加快医学影像处理、优化诊断并监测患者。
了解详情
制造业中的计算机视觉

制造业中的计算机视觉

利用 Ultralytics YOLO 模型优化制造流程。视觉 AI 可用于质量控制、缺陷检测、个人防护装备合规检查和装配线自动化。
了解详情
汽车领域的计算机视觉

汽车领域的计算机视觉

利用 Ultralytics YOLO 模型在汽车领域应用计算机视觉。视觉 AI 可提升道路安全、驾驶辅助和车辆自动化水平,让道路更智能。
了解详情
农业中的计算机视觉

农业中的计算机视觉

借助 Ultralytics YOLO 模型,将视觉 AI 引入智慧农业。为作物监测、牲畜追踪和精准农业提供支持,从而提高产量并实现更智能的生产。
了解详情

让我们携手构建 AI 的未来!

开启你的旅程,迎接机器学习的未来