ConvNeXt

Installation

ConvNeXt braucht kein optionales Extra. Alles, was es importiert, steckt in der Basisinstallation.

pip install libreyolo

Fine-Tuning mit Adaptern über lora=True ist die Ausnahme und braucht das Extra lora.

pip install "libreyolo[lora]"

Vorhersage

Die Gewichte werden beim ersten Aufruf von Hugging Face geladen und lokal zwischengespeichert.

from libreyolo import LibreYOLO, SAMPLE_IMAGE model = LibreYOLO("LibreConvNeXtt-cls.pt")result = model(SAMPLE_IMAGE, save=True) print(result.probs.top1, result.probs.top1conf)print(result.probs.top5)

libreyolo predict model=LibreConvNeXtt-cls.pt source=cat.jpg save=True

Das zurückgegebene Results-Objekt ist dasselbe, das jede Familie liefert, ein anderes Modell einzusetzen ist also eine Änderung von einer Zeile. Ein Klassifikator führt keine Boxen und keine Masken mit: result.probs enthält die Vorhersage für das ganze Bild, mit top1, top5, top1conf und top5conf. conf, iou und max_det werden aus Gründen der API-Parität akzeptiert, haben aber keine Wirkung, denn bei einem einzelnen Wahrscheinlichkeitsvektor gibt es nichts, worauf ein Schwellenwert oder eine Unterdrückung wirken könnte. Siehe Vorhersage für Quellen, Streaming und den Umgang mit Ergebnissen.

Varianten

Drei Größen, tiny/small/base, alle gleich trainiert und gleich ausgewertet, die Wahl ist also ein direkter Tausch von Parameterzahl gegen Accuracy. Die Aufgabe liegt fest: jede Größe deckt nur die Klassifikation ab. Der Name der Gewichtsdatei endet bei jeder Größe auf -cls.pt, und genau dieses Suffix liest die Factory, um an diese Familie zu routen; ein task=-Argument ist nicht nötig.

Training

Das Fine-Tuning startet vom veröffentlichten ImageNet-Backbone und baut die letzte Klassifikatorschicht automatisch auf die Klassenzahl des Zieldatensatzes um.

from libreyolo import LibreYOLO model = LibreYOLO("LibreConvNeXtt-cls.pt")model.train(data="imagenette160", epochs=5)

libreyolo train model=LibreConvNeXtt-cls.pt data=imagenette160 epochs=5

from libreyolo import LibreYOLO model = LibreYOLO("LibreConvNeXtt-cls.pt")model.train(data="imagenette160", epochs=5, lora=True)

libreyolo train model=LibreConvNeXtt-cls.pt data=imagenette160 \  epochs=50 device=0,1 batch=-1

Ohne weitere Angaben läuft das Training 100 Epochen mit lr0=1e-3, AdamW, einem Batch von 64 und Early Stopping nach 50 Epochen ohne Verbesserung. data akzeptiert ein Datensatzverzeichnis (train/ und val/, ein Ordner pro Klasse), einen bekannten Kurznamen wie imagenette160 oder eine .zip-URL. Die Blöcke von ConvNeXt enthalten die nn.Linear-MLPs, die LoRA braucht, deshalb wird lora=True hier unterstützt und schleust Adapter in die Block-MLPs ein, statt das ganze Backbone nachzutrainieren.

Siehe Training für Datensätze, Augmentierung, Multi-GPU und Logger.

Validierung

val() gibt ein Dictionary mit metrics/-Schlüsseln zurück. Bei der Klassifikation sind das die Top-1- und die Top-5-Accuracy über dem Validierungs-Split.

from libreyolo import LibreYOLO model = LibreYOLO("LibreConvNeXtt-cls.pt")metrics = model.val(data="imagenette160") print(metrics["metrics/accuracy_top1"])print(metrics["metrics/accuracy_top5"])

libreyolo val model=LibreConvNeXtt-cls.pt data=imagenette160

Export

Ein exportiertes Artefakt lädt über sein Dateisuffix wieder durch LibreYOLO(), eine .onnx- oder .engine-Datei verhält sich also wie ein Checkpoint und liefert dasselbe Results. Export listet die Argumente auf, die jedes Format akzeptiert, und die zusätzlichen, die einige davon mitbringen.

from libreyolo import LibreYOLO model = LibreYOLO("LibreConvNeXtt-cls.pt")model.export(format="onnx")model.export(format="tensorrt", half=True)

libreyolo export model=LibreConvNeXtt-cls.pt format=onnxlibreyolo export model=LibreConvNeXtt-cls.pt format=tensorrt half=True

from libreyolo import LibreYOLO, SAMPLE_IMAGE # Die Factory routet über das Dateisuffix: ein exportiertes Artefakt# lädt wie ein Checkpoint und liefert dasselbe Results-Objekt.model = LibreYOLO("LibreConvNeXtt-cls.onnx")result = model(SAMPLE_IMAGE) print(result.probs.top1)

Checkpoints

Jede veröffentlichte Gewichtsdatei dieser Familie.

Lizenzierung

Zitieren

@Article{liu2022convnet,
  author  = {Zhuang Liu and Hanzi Mao and Chao-Yuan Wu and Christoph Feichtenhofer and Trevor Darrell and Saining Xie},
  title   = {A ConvNet for the 2020s},
  journal = {Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR)},
  year    = {2022},
}