Depth8 数据集#
简介#
Ultralytics Depth8 数据集是一个紧凑的单目深度估计数据集,包含从SUN RGB-D数据集中采样的 8 张图像:4 张用于训练,4 张用于验证,来自其 Kinect v1 和 Kinect v2 采集数据(每个划分中每种传感器各 2 张),可提供密集且无伪影的真实深度图。它专为使用 YOLO26 深度估计模型和训练流程进行快速测试、调试与实验而设计——1.3 MB 的压缩包会在首次使用时自动下载,因此 yolo depth train data=depth8.yaml 可在几秒内开始训练。
访问 Ultralytics Platform 上的 Depth8,预览其 RGB-深度图像对并克隆该数据集用于训练。
Depth8 仅用于流程测试,不用于基准评测——其 8 张图像太少,无法得出有意义的深度指标。若要获得具有代表性的结果,请使用完整的 NYU Depth V2 或 SUN RGB-D 验证集。
数据集结构#
Depth8 遵循标准的 Ultralytics 深度数据集布局:RGB 图像与配对的 uint16 毫米深度 PNG(depth_scale: 1000)对应,并通过文件主干名进行匹配。
depth8-png/
├── images/
│ ├── train/ # 4 images
│ └── val/ # 4 images
└── depth/
├── train/ # 4 16-bit PNG depth maps
└── val/ # 4 16-bit PNG depth maps深度值来自真实的室内传感器采集数据,范围约为 0.5 m 至 4 m,完全处于完整 SUN RGB-D 数据集 ≤10 m 的范围内。
数据集 YAML#
Depth8 数据集配置定义在一个数据集 YAML 文件中,其中指定了数据集路径、类别名称以及小型打包子集的下载 URL。
# Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
# Depth8 dataset (8 clean-label indoor images from SUN RGB-D Kinect v1/v2, 4 train / 4 val) by Ultralytics
# Documentation: https://docs.ultralytics.com/datasets/depth/depth8
# Format: https://docs.ultralytics.com/datasets/depth#depth-map-format
# uint16 PNG values are divided by depth_scale to produce meters. The default 1000 gives 65,535
# one-millimeter depth values from 0.001 to 65.535 m; code 0 is invalid.
# Common values (resolution, uint16 maximum):
# ARKitScenes and NYU Depth V2: 1000 (1 mm, 65.535 m)
# KITTI: 256 (3.90625 mm, 255.996 m); Virtual KITTI 2: 100 (1 cm, 655.35 m)
# Example usage: yolo depth train data=depth8.yaml model=yolo26n-depth.pt
# parent
# ├── ultralytics
# └── datasets
# └── depth8-png ← downloads here (1.3 MB)
# ├── images/{train,val} # RGB images
# └── depth/{train,val} # paired 16-bit *.png depth maps (images/ -> depth/)
path: depth8-png # dataset root dir (relative to Ultralytics settings 'datasets_dir')
train: images/train # train images (relative to 'path') 4 images
val: images/val # val images (relative to 'path') 4 images
nc: 1
names:
0: depth
channels: 3
depth_scale: 1000 # PNG value 1000 = 1 meter
# Download script/URL (optional)
download: https://github.com/ultralytics/assets/releases/download/v0.0.0/depth8-png.zip使用方法#
要在 Depth8 数据集上以 640 的图像尺寸训练 YOLO26n-depth 模型,请使用以下示例。有关完整的训练选项列表,请参阅 YOLO 训练文档。
from ultralytics import YOLO
# Load a pretrained YOLO26n-depth model
model = YOLO("yolo26n-depth.pt")
# Train the model on Depth8
results = model.train(data="depth8.yaml", epochs=100, imgsz=640)引用和致谢#
Depth8 采样自 SUN RGB-D——有关来源归属和引用详情,请参阅完整的 SUN RGB-D 数据集页面。源数据集未指定许可证。
如果你在研究或开发工作中使用 SUN RGB-D 数据集,请引用以下论文:
@inproceedings{song2015sunrgbd,
title={SUN RGB-D: A RGB-D Scene Understanding Benchmark Suite},
author={Song, Shuran and Lichtenberg, Samuel P. and Xiao, Jianxiong},
booktitle={Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition (CVPR)},
year={2015}
}常见问题#
Ultralytics Depth8 数据集专为快速测试和调试单目深度估计流程而设计。它包含仅 8 张图像(训练集 4 张,验证集 4 张),打包在一个 1.3 MB 的自动下载压缩包中,可在几秒内验证完整的训练 / 验证 / 预测循环——包括配对深度图加载、数据增强、损失计算和指标计算,然后再扩展到 SUN RGB-D 或 NYU Depth V2 等完整数据集。
Depth8 从 SUN RGB-D 的 9,245-train/1,090-val 划分中采样 8 张图像,优先选择具有清晰、密集深度图的 Kinect v1/v2 采集数据。它使用完全相同的 16 位 PNG 深度格式,因此在 Depth8 上运行的流程无需修改即可在完整数据集上运行。与完整数据集不同,Depth8 可在几秒内完成下载,且不需要转换步骤。
不应该。Depth8 太小,无法进行有意义的模型比较,主要用于训练和评估流程检查。当你需要具有代表性的深度估计指标时,请使用完整的 NYU Depth V2 或 SUN RGB-D 验证集。



