Depth8 Dataset#

Einführung#

Das Ultralytics Depth8 Dataset ist ein kompaktes Dataset für die monokulare Tiefenschätzung mit 8 Bildern, die aus dem SUN RGB-D Dataset entnommen wurden: 4 für das Training und 4 für die Validierung. Sie stammen aus dessen Kinect v1- und Kinect v2-Aufnahmen (zwei pro Sensor in jedem Split) für dichte, artefaktfreie Ground-Truth-Tiefenkarten. Es wurde für das schnelle Testen, Debuggen und Experimentieren mit YOLO26 Tiefenschätzungsmodellen und Training-Pipelines entwickelt – das 1,3 MB große Archiv wird bei der ersten Verwendung automatisch heruntergeladen, sodass yolo depth train data=depth8.yaml innerhalb von Sekunden mit dem Training beginnt.

Datensatzstruktur#

Depth8 folgt dem Standard-Layout für Ultralytics-Tiefen-Datasets: RGB-Bilder mit gepaarten .npy float32-Tiefenkarten in Metern, abgeglichen über den Dateinamen.

depth8/
├── images/
│   ├── train/  # 4 images
│   └── val/    # 4 images
└── depth/
    ├── train/  # 4 float32 .npy depth maps
    └── val/    # 4 float32 .npy depth maps

Tiefenwerte sind echte Sensoraufnahmen aus Innenräumen, die von etwa 0,5 m bis 4 m reichen, was gut innerhalb des ≤10 m Bereichs des vollständigen SUN RGB-D-Datasets liegt.

Datensatz-YAML#

Die Konfiguration des Depth8-Datasets ist in einer Dataset-YAML-Datei definiert, die Dataset-Pfade, Klassennamen und die Download-URL für das kleine Paket-Subset angibt.

# 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
# Example usage: yolo depth train data=depth8.yaml model=yolo26n-depth.pt
# parent
# ├── ultralytics
# └── datasets
#     └── depth8 ← downloads here (1.3 MB)
#         ├── images/{train,val}  # RGB images
#         └── depth/{train,val}   # paired *.npy, float32 meters (images/ -> depth/)

path: depth8 # 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

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

Verwendung#

Um ein YOLO26n-depth Modell mit dem Depth8 Dataset und einer Bildgröße von 640 zu trainieren, verwende die folgenden Beispiele. Eine vollständige Liste der Trainingsoptionen findest du in der YOLO-Trainingsdokumentation.

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)

Zitate und Danksagungen#

Depth8 wurde aus SUN RGB-D entnommen – Details zur Lizenz findest du auf der vollständigen SUN RGB-D Dataset-Seite.

Wenn du das SUN RGB-D-Dataset in deiner Forschung oder Entwicklungsarbeit verwendest, zitiere bitte das folgende Paper:

@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}
}

FAQ#