Набор данных Cityscapes8

Введение

Набор данных Ultralytics Cityscapes8 — это компактный набор данных для семантической сегментации, состоящий из 8 изображений, взятых из набора данных Cityscapes: 4 для обучения и 4 для валидации. Он предназначен для быстрой проверки, отладки и экспериментов с моделями семантической сегментации YOLO и конвейерами обучения. Его содержимое с городскими сценами позволяет полезно протестировать конвейер перед переходом к полному набору данных Cityscapes.

Cityscapes8 использует те же 19 классов оценки и то же поведение label_mapping, что и полный набор данных Cityscapes, и полностью совместим с рабочими процессами семантической сегментации YOLO26.

YAML набора данных

Конфигурация набора данных Cityscapes8 определена в YAML-файле, который задает пути к данным, имена классов и другие важные метаданные. Ты можешь ознакомиться с официальным файлом cityscapes8.yaml в репозитории Ultralytics на GitHub. YAML-файл содержит URL для скачивания небольшого упакованного подмножества.

# Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license

# Cityscapes semantic segmentation dataset (19 classes)
# Documentation: https://docs.ultralytics.com/datasets/semantic/cityscapes8/
# Example usage: yolo semantic train data=cityscapes8.yaml model=yolo26n-sem.pt
# parent
# ├── ultralytics
# └── datasets
#     └── cityscapes8 ← downloads here (small subset)
#         └── images
#         └── masks

# Dataset root directory
path: cityscapes8 # dataset root dir
train: images/train # train images (relative to 'path') 4 images
val: images/val # val images (relative to 'path') 4 images

masks_dir: masks # semantic mask directory

# Cityscapes 19-class labels
names:
  0: road
  1: sidewalk
  2: building
  3: wall
  4: fence
  5: pole
  6: traffic light
  7: traffic sign
  8: vegetation
  9: terrain
  10: sky
  11: person
  12: rider
  13: car
  14: truck
  15: bus
  16: train
  17: motorcycle
  18: bicycle

# Map source label IDs to train IDs; ignore_label is converted to 255.
label_mapping:
  -1: ignore_label
  0: ignore_label
  1: ignore_label
  2: ignore_label
  3: ignore_label
  4: ignore_label
  5: ignore_label
  6: ignore_label
  7: 0
  8: 1
  9: ignore_label
  10: ignore_label
  11: 2
  12: 3
  13: 4
  14: ignore_label
  15: ignore_label
  16: ignore_label
  17: 5
  18: ignore_label
  19: 6
  20: 7
  21: 8
  22: 9
  23: 10
  24: 11
  25: 12
  26: 13
  27: 14
  28: 15
  29: ignore_label
  30: ignore_label
  31: 16
  32: 17
  33: 18

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

Использование

Чтобы обучить модель YOLO26n-sem на наборе данных Cityscapes8 в течение 100 эпох с размером изображения 1024, используй следующие примеры. Полный список опций обучения смотри в документации по обучению YOLO.

from ultralytics import YOLO

# Load a pretrained YOLO26n-sem model
model = YOLO("yolo26n-sem.pt")

# Train the model on Cityscapes8
results = model.train(data="cityscapes8.yaml", epochs=100, imgsz=1024)

Цитирование и благодарности

Если ты используешь набор данных Cityscapes в своих исследованиях или разработках, пожалуйста, процитируй следующую статью:

@inproceedings{Cordts2016Cityscapes,
  title={The Cityscapes Dataset for Semantic Urban Scene Understanding},
  author={Cordts, Marius and Omran, Mohamed and Ramos, Sebastian and Rehfeld, Timo and Enzweiler, Markus and Benenson, Rodrigo and Franke, Uwe and Roth, Stefan and Schiele, Bernt},
  booktitle={Proc. of the IEEE Conference on Computer Vision and Pattern Recognition (CVPR)},
  year={2016}
}

Особая благодарность команде Cityscapes за их постоянный вклад в развитие технологий автономного вождения и сообщества компьютерного зрения.

Часто задаваемые вопросы (FAQ)

Для чего используется набор данных Ultralytics Cityscapes8?

Набор данных Ultralytics Cityscapes8 предназначен для быстрой проверки и отладки моделей семантической сегментации. Содержа всего 8 изображений (4 для обучения, 4 для валидации), он идеально подходит для верификации конвейеров семантической сегментации YOLO, включая загрузку масок, аугментации, валидацию и пути экспорта, перед масштабированием до полного набора данных Cityscapes. Ознакомься с YAML-конфигурацией Cityscapes8 для получения подробной информации.

Как обучить модель YOLO26, используя набор данных Cityscapes8?

Ты можешь обучить модель семантической сегментации YOLO26 на Cityscapes8, используя Python или CLI:

from ultralytics import YOLO

# Load a pretrained YOLO26n-sem model
model = YOLO("yolo26n-sem.pt")

# Train the model on Cityscapes8
results = model.train(data="cityscapes8.yaml", epochs=100, imgsz=1024)

Для дополнительных опций обучения обратись к документации по обучению YOLO.

Стоит ли использовать Cityscapes8 для бенчмаркинга?

Нет. Cityscapes8 слишком мал для значимого сравнения моделей и предназначен только для проверки конвейеров обучения и оценки. Используй полный набор валидационных данных Cityscapes, когда тебе нужны репрезентативные результаты бенчмаркинга для семантической сегментации.