Ultralytics YOLO27:

Датасет Cityscapes8#

Введение#

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

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

Примечание

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

Структура датасета#

Cityscapes8 повторяет структуру полного датасета, но не содержит разбиения test:

cityscapes8/
├── images/
│   ├── train/  # 4 images
│   └── val/    # 4 images
└── masks/
    ├── train/  # 4 single-channel PNG masks
    └── val/    # 4 single-channel PNG masks

Маски сопоставляются с изображениями через поле masks_dir: masks, а label_mapping преобразует исходные идентификаторы меток Cityscapes в 19 последовательных идентификаторов для обучения, описанных в полной структуре датасета Cityscapes.

YAML датасета#

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

ultralytics/cfg/datasets/cityscapes8.yaml
# 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)

Ты также можешь управлять Cityscapes8 и обучать модели семантической сегментации в облаке с помощью Ultralytics Platform.

Цитирование, лицензия и благодарности#

Cityscapes8 является поднабором Cityscapes, который распространяется по специальной некоммерческой лицензии. Подробности смотри на полной странице датасета Cityscapes.

Если ты используешь датасет 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 за её постоянный вклад в развитие сообществ автономного вождения и компьютерного зрения.

Часто задаваемые вопросы#

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

  • Cityscapes8 содержит 8 изображений (4 для обучения и 4 для валидации), выбранных из разбиения Cityscapes на 2 975 изображений для обучения и 500 для валидации, и использует те же 19 классов и label_mapping, поэтому конвейер, работающий с Cityscapes8, запускается на полном датасете без изменений — достаточно указать в data= путь к cityscapes.yaml вместо cityscapes8.yaml. В отличие от полного датасета, Cityscapes8 не требует ручного скачивания и не содержит разбиения test.

  • Ты можешь обучить модель семантической сегментации 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 слишком мал для содержательного сравнения моделей и предназначен для проверки конвейеров обучения и оценки. Если тебе нужны репрезентативные результаты бенчмаркинга семантической сегментации, используй полный набор валидационных данных Cityscapes.

Комментарии