Набор данных Cityscapes8#
Введение#
Набор данных Cityscapes8 от Ultralytics — это компактный набор данных для семантической сегментации, состоящий из 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 преобразует исходные ID меток Cityscapes в 19 последовательных ID классов для обучения, описанных в структуре полного набора данных Cityscapes.
Файл 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
# Загрузи предварительно обученную модель YOLO26n-sem
model = YOLO("yolo26n-sem.pt")
# Обучи модель на 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 за её постоянный вклад в сообщества автономного вождения и компьютерного зрения.
Часто задаваемые вопросы#
Набор данных Cityscapes8 от Ultralytics предназначен для быстрого тестирования и отладки моделей семантической сегментации. В нём всего 8 изображений (4 для обучения, 4 для валидации), поэтому он идеально подходит для проверки конвейеров семантической сегментации YOLO, включая загрузку масок, аугментации, валидацию и экспорт, прежде чем переходить к полному набору данных Cityscapes. Дополнительную информацию смотри в конфигурации YAML Cityscapes8.
Cityscapes8 включает 8 изображений (4 для обучения, 4 для валидации) из выборки Cityscapes, содержащей 2 975 обучающих и 500 проверочных изображений. В нём используются те же 19 классов и
label_mapping, поэтому конвейер, работающий на Cityscapes8, без изменений будет работать и на полном наборе данных — просто укажиdata=вместоcityscapes8.yamlвcityscapes.yaml. В отличие от полного набора данных, Cityscapes8 не нужно скачивать вручную и в нём нет разделаtest.Ты можешь обучить модель YOLO26 для семантической сегментации на Cityscapes8 с помощью Python или CLI:
Пример обученияfrom ultralytics import YOLO # Загрузи предварительно обученную модель YOLO26n-sem model = YOLO("yolo26n-sem.pt") # Обучи модель на Cityscapes8 results = model.train(data="cityscapes8.yaml", epochs=100, imgsz=1024)Дополнительные параметры обучения см. в документации по обучению YOLO.
Нет. Cityscapes8 слишком мал для содержательного сравнения моделей и предназначен для проверки конвейеров обучения и оценки. Если нужны репрезентативные результаты бенчмарка семантической сегментации, используй полную проверочную выборку Cityscapes.



