Matrice d'augmentation

Réglages

Il s'agit de noms de champs TrainConfig, et non de leurs formes dans le CLI. Le CLI associe ses propres alias à ces champs, --mosaic définit donc mosaic_prob.

Les quatre derniers constituent le groupe de classification. Les familles de détection les ignorent. mixup est un réglage réservé à l'API\u00a0: l'option --mixup du CLI est l'alias du réglage de détection mixup_prob.

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 style DETR utilise une transformation directe sans mosaic. Sa distorsion photométrique, son zoom arrière et son recadrage IoU sont des constantes de la recette plutôt que des réglages configurables. hsv_prob et les réglages géométriques ne l'atteignent donc jamais. Le pipeline de classification utilise une transformation ImageFolder dont le retournement horizontal est fixé à 0.5 plutôt que défini par flip_prob. La variation d'échelle sémantique et le HSV proviennent d'attributs de classe des familles, et non des réglages de configuration. Les retournements de restauration sont des opérations couplées sur l'entrée et la cible, avec une probabilité fixe de 0.5.

no_aug_epochs est respecté partout, mais les éléments désactivés diffèrent\u00a0: mosaic et MixUp pour le style YOLOX, la transformation affine et MixUp pour YOLO-NAS, les fortes augmentations photométriques et de recadrage ainsi que la fin du learning rate pour le style DETR, et la fin du scheduler pour les autres.

Familles par archétype

Vingt-cinq 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\u00a0; 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.