Объяснение размера обучения

YOLOv6 поддерживает три режима размера обучения.

1. Квадратная форма обучения

Если вы передаете только одно число в --img-size, например --img-size 640, то более длинная сторона изображения будет сохранять пропорции и масштабироваться до 640, а более короткая сторона будет масштабирована с тем же коэффициентом, затем заполнена до 640. Изображение, отправленное в модель, будет иметь разрешение (640, 640, 3).

2. Прямоугольная форма обучения

Если вы передаете --img-size 640 и --rect, то более длинная сторона изображения будет сохранять пропорции и масштабироваться до 640, а более короткая сторона будет масштабирована с тем же коэффициентом, затем заполнена до кратного 32 (если необходимо). Например, если форма одного изображения (720, 1280, 3), после сохранения пропорций и масштабирования, его форма изменится на (360, 640, 3), однако 320 не является кратным 32, поэтому оно будет заполнено до (384, 640, 3).

3. Конкретная форма

В режиме прямоугольной формы обучение может иметь разные размеры обучения, такие как (1080, 1920, 3) и (1200, 1600, 3). Если вы хотите указать конкретную форму, вы можете использовать команду --specific-shape и указать ваш размер обучения с помощью --height и --width, например:

python tools/train.py --data data/dataset.yaml --conf configs/yolov6n.py --specific-shape --width 1920 --height 1080

Тогда разрешение обучающих данных будет (1080, 1920, 3) независимо от формы изображения в наборе данных.