Files

1.3 KiB

Пример структуры датасета

Эталонный пример организации данных для обучения YOLO.

Файлы

example/
├── images/
│   ├── slice_0001.png
│   └── slice_0002.png
├── images_labels/
│   ├── slice_0001.txt          <- bbox для детекции
│   ├── slice_0002.txt
│   └── polygons/
│       ├── slice_0001.txt      <- полигоны для сегментации
│       └── slice_0002.txt
└── images_result/
    ├── slice_0001_preview.png
    └── slice_0002_preview.png

Формат детекции (images_labels/slice_XXXX.txt)

Каждая строка = один объект:

<class_id> <x_center> <y_center> <width> <height>

Пример:

0 0.512 0.634 0.150 0.200
1 0.300 0.450 0.100 0.080

Формат сегментации (images_labels/polygons/slice_XXXX.txt)

Каждая строка = полигон одного объекта:

<class_id> <x1> <y1> <x2> <y2> ... <xn> <yn>

Пример:

0 0.450 0.500 0.460 0.510 0.470 0.505 0.465 0.495

Все координаты нормализованы к [0, 1] относительно размеров изображения.