Ultralytics YOLO27:

KITTI 数据集#

Open KITTI Dataset In Colab

Ultralytics KITTI 数据集是一个用于自动驾驶的 2D 目标检测数据集,包含 7,481 张标注图像(5,985 张用于训练,1,496 张用于验证),涵盖 8 个类别——car、van、truck、pedestrian、person_sitting、cyclist、tram 和 misc。该数据集由卡尔斯鲁厄理工学院和芝加哥丰田技术学院发布,图像来自真实世界的城市、乡村和高速公路驾驶场景。



Watch: How to Train Ultralytics YOLO on the KITTI Dataset | Object Detection, Inference & ONNX Export 🚀

更广泛的 KITTI Vision Benchmark Suite 还涵盖深度估计、光流、立体视觉和视觉里程计,但此处的 Ultralytics kitti.yaml 配置针对 2D 目标检测进行了设置,并完全兼容 Ultralytics YOLO26

数据集结构#

警告

这里不包含原始 KITTI 测试集,因为它没有公开的真实标注。

该数据集包含 7,481 张标注图像,涵盖汽车、行人和骑行者等对象,并根据 kitti.yaml 配置划分为两个预定义子集:

划分图像描述
训练5,985用于模型训练的标注图像
验证1,496用于评估和基准测试的留出图像

目标类别#

kitti.yaml 文件定义了 8 个目标类别,涵盖驾驶场景中常见的车辆、人员和其他道路使用者:

KITTI 类别
  1. car
  2. van
  3. truck
  4. pedestrian
  5. person_sitting
  6. cyclist
  7. tram
  8. misc

应用#

KITTI 数据集支持自动驾驶和机器人领域的一系列 2D 检测应用:

  • 自动驾驶车辆感知:训练模型检测和跟踪汽车、行人及骑行者,帮助自动驾驶系统安全导航。
  • ADAS 开发:基于真实驾驶影像构建碰撞预警和行人检测等驾驶辅助功能。
  • 交通与道路场景分析:检测并统计车辆和道路使用者,以研究交通流量和道路安全。
  • 计算机视觉基准测试:使用 KITTI 作为标准基准,评估 2D 目标检测跟踪模型。

要在浏览器中标注自己的驾驶图像、进行训练并管理数据集版本,请通过 Ultralytics Platform运行完整工作流。

数据集 YAML#

Ultralytics 使用 YAML 文件定义 KITTI 数据集配置。该文件指定了训练所需的数据集路径、类别标签和元数据。配置文件位于 https://github.com/ultralytics/ultralytics/blob/main/ultralytics/cfg/datasets/kitti.yaml

ultralytics/cfg/datasets/kitti.yaml
# Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license

# KITTI dataset by Karlsruhe Institute of Technology and Toyota Technological Institute at Chicago
# Documentation: https://docs.ultralytics.com/datasets/detect/kitti
# Example usage: yolo train data=kitti.yaml
# parent
# ├── ultralytics
# └── datasets
#     └── kitti ← downloads here (390.5 MB)

# Train/val/test sets as 1) dir: path/to/imgs, 2) file: path/to/imgs.txt, or 3) list: [path/to/imgs1, path/to/imgs2, ..]
path: kitti # dataset root dir
train: images/train # train images (relative to 'path') 5985 images
val: images/val # val images (relative to 'path') 1496 images

names:
  0: car
  1: van
  2: truck
  3: pedestrian
  4: person_sitting
  5: cyclist
  6: tram
  7: misc

# Download script/URL (optional)
download: https://github.com/ultralytics/assets/releases/download/v0.0.0/kitti.zip

使用方法#

要在 KITTI 数据集上训练 YOLO26n 模型 100 个轮次,并使用 640 的图像尺寸,请使用以下命令。数据集(390.5 MB)会在首次使用时自动下载。更多详情请参阅训练页面。

训练示例
from ultralytics import YOLO

# Load a pretrained YOLO26 model
model = YOLO("yolo26n.pt")

# Train on kitti dataset
results = model.train(data="kitti.yaml", epochs=100, imgsz=640)

你还可以使用同一个配置文件,直接从命令行或 Python API 执行评估、推理导出任务。

示例图像和标注#

下面的示例展示了数据集中的一个驾驶场景及其 2D 边界框标注。KITTI 图像涵盖在真实交通环境中采集的城市、乡村和高速公路场景,为模型提供了多样的目标尺度、视角和光照条件。

KITTI dataset vehicle detection sample

引用和致谢#

如果你在研究中使用 KITTI 数据集,请引用以下论文:

引用
@article{Geiger2013IJRR,
  author = {Andreas Geiger and Philip Lenz and Christoph Stiller and Raquel Urtasun},
  title = {Vision meets Robotics: The KITTI Dataset},
  journal = {International Journal of Robotics Research (IJRR)},
  year = {2013}
}

我们感谢 KITTI Vision Benchmark Suite 提供这一全面的数据集,它持续推动计算机视觉、机器人和自动驾驶系统的发展。更多信息请访问 KITTI 网站

常见问题#

  • Ultralytics KITTI 数据集用于训练和评估自动驾驶领域的 2D 目标检测模型。它提供了跨越 8 个类别的 7,481 张标注图像,包括汽车、行人和骑行者,并广泛用于感知模型的基准测试。

  • Ultralytics KITTI 配置包含 7,481 张图像——5,985 张用于训练,1,496 张用于验证——没有单独的测试集。每张图像都标注了 8 个类别:car、van、truck、pedestrian、person_sitting、cyclist、tram 和 misc。

  • 不包含。Ultralytics KITTI 配置仅提供训练集(5,985 张图像)和验证集(1,496 张图像)。原始 KITTI 测试集被排除在外,因为它没有公开的真实标注。

  • 首次使用 data="kitti.yaml" 进行训练时,数据集(390.5 MB)会自动下载,无需手动操作。Ultralytics 会获取图像和标签,并将其解压到本地 datasets 目录。你可以在检测数据集概览中浏览相关数据集。

  • 可以,KITTI 完全兼容 Ultralytics YOLO26。你可以直接使用提供的 YAML 配置文件训练验证模型。

  • 你可以在 https://github.com/ultralytics/ultralytics/blob/main/ultralytics/cfg/datasets/kitti.yaml访问 kitti.yaml 文件。该文件定义了数据集路径以及训练所用的 8 个类别名称。

评论