Датасет Depth8#
Введение#
Датасет Depth8 от Ultralytics — это компактный датасет для оценки глубины по одному изображению, содержащий 8 изображений из датасета SUN RGB-D: 4 для обучения и 4 для валидации. Для получения плотных карт эталонной глубины без артефактов выбраны кадры, снятые датчиками Kinect v1 и Kinect v2 (по два кадра с каждого датчика в каждой выборке). Датасет предназначен для быстрого тестирования, отладки и экспериментов с моделями YOLO26 для оценки глубины и конвейерами обучения: архив размером 1,3 МБ загружается автоматически при первом использовании, поэтому yolo depth train data=depth8.yaml запускает обучение за считаные секунды.
Открой Depth8 на платформе Ultralytics, чтобы просмотреть пары RGB и глубины и клонировать датасет для обучения.
Depth8 предназначен только для тестирования конвейеров, а не для бенчмаркинга: 8 изображений слишком мало для получения содержательных метрик глубины. Для репрезентативных результатов используй полные валидационные выборки NYU Depth V2 или SUN RGB-D.
Структура набора данных#
Depth8 соответствует стандартной структуре датасета глубины Ultralytics: изображения RGB сопоставлены с PNG-файлами uint16 с глубиной в миллиметрах (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 до 4 м, что укладывается в диапазон ≤10 м полного датасета SUN RGB-D.
Файл 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Использование#
Чтобы обучить модель YOLO26n-depth на датасете Depth8 с размером изображения 640, используй следующие примеры. Полный список параметров обучения см. в документации по обучению YOLO.
from ultralytics import YOLO
# Загрузить предварительно обученную модель YOLO26n-depth
model = YOLO("yolo26n-depth.pt")
# Обучить модель на 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 предназначен для быстрого тестирования и отладки конвейеров оценки глубины по одному изображению. Архив размером 1,3 МБ, который автоматически скачивается и содержит всего 8 изображений (4 для обучения, 4 для валидации), позволяет за секунды проверить полный цикл train / val / predict — загрузку пар с картами глубины, аугментацию, вычисление функции потерь и метрики — перед переходом к полному датасету, например SUN RGB-D или NYU Depth V2.
В Depth8 выбрано 8 изображений из выборок SUN RGB-D, содержащих 9 245 изображений для обучения и 1 090 для валидации; предпочтение отдано кадрам Kinect v1/v2 с чистыми плотными картами глубины. Используется тот же формат глубины PNG с разрядностью 16 бит, поэтому конвейер, работающий с Depth8, будет работать с полным датасетом без изменений. В отличие от полного датасета, Depth8 скачивается за секунды и не требует преобразования.
Нет. Depth8 слишком мал для содержательного сравнения моделей и предназначен для проверки конвейеров обучения и оценки. Для получения репрезентативных метрик оценки глубины используй полные валидационные выборки NYU Depth V2 или SUN RGB-D.



