Matrice d'augmentation

Réglages

Il s'agit de noms de champs TrainConfig, et non de leurs formes dans le CLI. train() et le CLI associent les formes courtes à ces champs, mosaic définit donc mosaic_prob.

Les quatre derniers forment les contrôles de classification. Les familles de détection les ignorent. train() et la CLI routent mixup vers le mélange de lots pour les classifieurs et vers mixup_prob pour les détecteurs.

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"))

Trois états

ignored est l'état à vérifier avant une exécution, car aucune erreur ne se produit. Le CLI avertit lorsqu'un paramètre d'entraînement explicitement défini est ignoré par la famille sélectionnée. Le trainer avertit lorsque mixup_prob > 0 ne peut pas se déclencher, car la famille conditionne MixUp à mosaic et mosaic_prob vaut zéro.

Archétypes de pipelines

Chaque famille couverte suit l'un de six pipelines, avec quelques écarts par famille énumérés ci-dessous.

Dans le pipeline de style YOLOX, le prétraitement par échantillon applique les variations HSV et les retournements, tandis que la transformation affine et MixUp s'exécutent uniquement dans la branche mosaic. YOLO-NAS exécute plutôt une transformation affine par échantillon toujours active, ignore mosaic et applique MixUp indépendamment, en réutilisant mosaic_scale comme plage d'échelle affine.

Le pipeline de type DETR est une transformation sans mosaic. La distorsion photométrique, le zoom-out et le recadrage par IoU sont des constantes de recette, pas des paramètres configurables : hsv_prob et les contrôles géométriques ne les pilotent donc pas. La classification utilise flip_prob pour les retournements horizontaux et flipud pour les verticaux. Les variations d'échelle et HSV de segmentation sémantique viennent d'attributs de la classe du modèle ; les retournements de restauration s'appliquent conjointement à l'entrée et à la cible avec une probabilité fixe de 0.5.

no_aug_epochs est pris en compte partout, mais désactive des opérations différentes : mosaic et MixUp pour les pipelines de type YOLOX ; la transformation affine et MixUp pour YOLO-NAS ; les augmentations photométriques fortes et le recadrage, avec la fin du planning du taux d'apprentissage, pour DETR ; l'augmentation automatique, erasing, MixUp et CutMix pour la classification. Le recadrage et les retournements de classification restent actifs.

Familles par archétype

Trente et une familles sont couvertes. Une famille absente de cette liste renvoie un ensemble ignored vide, aucun avertissement n'est donc émis à son sujet.

Écarts

ec et dinov2 sont des familles multitâches. Un réglage n'est donc marqué ignored que lorsque chaque tâche entraînable de la famille l'ignore. Un avertissement du CLI n'est ainsi jamais erroné pour une tâche tout en étant correct pour une autre.

Dome-DETR hérite sans modification des transformations de D-FINE. La seule fonctionnalité qu'il ne peut pas accepter est l'entraînement multi-échelle, que sa configuration désactive plutôt que la spécification d'augmentation.

Réglages propres aux familles

Certaines familles portent des réglages d'augmentation dans leur propre sous-classe TrainConfig plutôt que dans la classe de base. Le CLI ne les expose pas ; définissez-les par l'API Python.

rot90 s'applique à detect et OBB sur yolo9.

Interroger la spécification

Un objet Support est un tuple nommé contenant status et note, cette dernière explique pourquoi un réglage est ignoré ou conditionné pour cette famille.

Condition mosaic

Pour une famille de style YOLOX, associer mixup_prob=0.5 à mosaic_prob=0 désactive entièrement MixUp, car celui-ci ne s'applique qu'aux échantillons mosaic. Cette combinaison survient facilement lorsque mosaic est désactivé en fin d'entraînement. Le trainer consigne un avertissement qui nomme la famille, et mixup_gating_warning est la fonction pure qui le produit.