Augmentationsmatrix

Einstellungen

Dies sind Feldnamen von TrainConfig, nicht ihre Schreibweisen in der CLI. Die CLI bildet eigene Aliasse darauf ab. --mosaic setzt daher mosaic_prob.

Die letzten vier bilden das Klassifizierungspaket. Erkennungsfamilien ignorieren sie. mixup ist ausschließlich über die API verfügbar. Das CLI-Argument --mixup ist der Alias für das bei der Erkennung verwendete 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"))

Drei Statuswerte

Prüfe vor einem Lauf insbesondere ignored, da kein Fehler ausgelöst wird. Die CLI warnt, wenn ein ausdrücklich gesetzter Trainingsparameter von der gewählten Familie ignoriert wird. Der Trainer warnt außerdem, wenn mixup_prob > 0 nicht ausgelöst werden kann, weil die Familie MixUp an Mosaic koppelt und mosaic_prob null ist.

Pipeline-Archetypen

Jede erfasste Familie folgt einer von sechs Pipelines. Einige familienbezogene Abweichungen stehen weiter unten.

In der Pipeline im YOLOX-Stil wendet die Vorverarbeitung je Sample HSV-Jitter und Spiegelungen an. Die affine Transformation und MixUp laufen nur innerhalb des Mosaic-Zweigs. YOLO-NAS führt dagegen immer eine affine Transformation je Sample aus, ignoriert Mosaic und wendet MixUp unabhängig an. Dabei wird mosaic_scale als Skalierungsbereich der affinen Transformation wiederverwendet.

Die Pipeline im DETR-Stil verwendet eine unveränderte Durchleitung ohne Mosaic. Fotometrische Verzerrung, Zoom-out und IoU-Beschnitt sind feste Rezeptwerte und keine konfigurierbaren Einstellungen. Deshalb erreichen hsv_prob und die Geometrieeinstellungen diese Pipeline nie. Die Klassifizierungspipeline verwendet eine ImageFolder-Transformation mit einer festen Wahrscheinlichkeit von 0,5 für horizontales Spiegeln statt flip_prob. Bei der semantischen Segmentierung stammen Skalierungs-Jitter und HSV aus Klassenattributen der Familie statt aus Konfigurationseinstellungen. Restauration koppelt die Spiegelungen von Eingabe und Ziel mit einer festen Wahrscheinlichkeit von 0,5.

no_aug_epochs wird überall berücksichtigt, schaltet aber unterschiedliche Elemente ab. Im YOLOX-Stil betrifft dies Mosaic und MixUp, bei YOLO-NAS die affine Transformation und MixUp, im DETR-Stil die starken fotometrischen und Beschnitt-Augmentationen sowie das Ende des Lernratenverlaufs und bei den übrigen Pipelines das Ende des Schedulers.

Familien nach Archetyp

Fünfundzwanzig Familien werden erfasst. Für eine Familie außerhalb dieser Liste wird eine leere Menge ignorierter Einstellungen zurückgegeben, sodass keine Warnung erscheint.

Abweichungen

ec und dinov2 sind Multi-Task-Familien. Eine Einstellung wird deshalb nur dann als ignoriert markiert, wenn jede trainierbare Aufgabe der Familie sie ignoriert. So kann die CLI-Warnung nie für eine Aufgabe falsch und für eine andere richtig sein.

Dome-DETR übernimmt die Transformationen von D-FINE unverändert. Nur Multi-Scale-Training wird nicht unterstützt. Dies wird von seiner Konfiguration statt der Augmentationsspezifikation deaktiviert.

Familienspezifische Einstellungen

Einige Familien definieren Augmentationseinstellungen in ihrer eigenen TrainConfig-Unterklasse statt in der Basisklasse. Die CLI stellt sie nicht bereit. Lege sie über die Python-API fest.

rot90 gilt bei yolo9 für detect und OBB.

Abfrage der Spezifikation

Support ist ein benanntes Tupel aus status und note. Die Anmerkung erklärt, warum eine Einstellung bei dieser Familie ignoriert oder gekoppelt wird.

Mosaic-Kopplung

Bei einer Familie im YOLOX-Stil deaktiviert mixup_prob=0.5 zusammen mit mosaic_prob=0 MixUp vollständig, da MixUp nur auf Mosaic-Samples angewendet wird. Diese Kombination entsteht leicht, wenn Mosaic spät im Training deaktiviert wird. Der Trainer protokolliert eine Warnung mit dem Familiennamen. mixup_gating_warning ist die zugrunde liegende reine Funktion.