Matriz de data augmentation

Os parâmetros

Estes são nomes de campos do TrainConfig, não a grafia da CLI. A CLI mapeia seus próprios aliases sobre eles, então --mosaic define mosaic_prob.

Os quatro últimos formam o pacote de classificação. As famílias de detecção os ignoram. mixup é um parâmetro exclusivo da API: na CLI, --mixup é o alias do mixup_prob de detecção.

from libreyolo.data.augment.spec import (    AUG_KNOBS,    aug_support,    ignored_aug_params,    uses_mosaic_gating,) print(sorted(AUG_KNOBS)) table = aug_support("yolo9")print(table["mixup_prob"].status, table["mixup_prob"].note) print(sorted(ignored_aug_params("dfine")))print(uses_mosaic_gating("yolo9"), uses_mosaic_gating("yolonas"))

Os três status

ignored é o que vale conferir antes de uma execução, porque nada falha. A CLI avisa quando um parâmetro de treinamento definido explicitamente é um dos que a família selecionada ignora, e o treinador avisa quando mixup_prob > 0 não pode disparar porque a família condiciona o MixUp ao mosaico e mosaic_prob é zero.

Arquétipos de pipeline

Toda família coberta segue um de seis pipelines, com um punhado de desvios por família listados abaixo.

No pipeline no estilo YOLOX, o pré-processamento por amostra aplica o jitter HSV e os flips, enquanto o warp afim e o MixUp rodam apenas dentro do ramo do mosaico. Já o YOLO-NAS roda um afim por amostra que está sempre ligado, ignora o mosaico e aplica o MixUp de forma independente, reaproveitando mosaic_scale como faixa de escala do afim.

O pipeline no estilo DETR é um transform de passagem, sem mosaico. Sua distorção fotométrica, o zoom-out e o crop por IoU são constantes da receita, e não parâmetros configuráveis, e é por isso que hsv_prob e os parâmetros de geometria nunca chegam até ele. O pipeline de classificação usa um transform de ImageFolder cujo flip horizontal é um 0.5 fixo, e não flip_prob. O jitter de escala e o HSV da semântica vêm de atributos de classe da família, e não de parâmetros de configuração, e os flips da restauração são operações acopladas de entrada e alvo, com probabilidade fixa de 0.5.

no_aug_epochs é respeitado em todo lugar, embora o que ele desliga varie: mosaico e MixUp no estilo YOLOX, o afim e o MixUp no YOLO-NAS, as augmentations fotométricas e de crop fortes mais a cauda do learning rate no estilo DETR, e a cauda do scheduler no resto.

Famílias por arquétipo

Vinte e cinco famílias são cobertas. Uma família fora desta lista retorna um conjunto de ignorados vazio, então nenhum aviso é emitido para ela.

Desvios

ec e dinov2 são famílias multitarefa, então um parâmetro só é marcado como ignorado quando todas as tarefas treináveis da família o ignoram. Isso impede que o aviso da CLI seja errado para uma tarefa enquanto está certo para outra.

O Dome-DETR herda os transforms do D-FINE sem alterações. A única coisa que ele não aceita é o treinamento multiescala, que sua configuração desativa, e não a spec de augmentation.

Parâmetros específicos de família

Algumas famílias carregam parâmetros de data augmentation na sua própria subclasse de TrainConfig, e não na base. A CLI não expõe esses parâmetros; defina-os pela API Python.

rot90 se aplica a detect e OBB no yolo9.

Consultando a spec

Um Support é uma tupla nomeada de status e note, em que a nota explica por que um parâmetro é ignorado ou condicionado para aquela família.

O gate do mosaico

Para uma família no estilo YOLOX, mixup_prob=0.5 com mosaic_prob=0 desativa o MixUp por completo, porque o MixUp se aplica apenas a amostras de mosaico. Essa combinação é fácil de alcançar ao desligar o mosaico no fim do treinamento. O treinador registra um aviso nomeando a família, e mixup_gating_warning é a função pura por trás dele.