ConvNeXt V2

ConvNeXt V2 is an image classifier with global response normalization.

Aufgaben
classification
Installation
pip install libreyolo
Supportstufe
Unterstützt, seit v1.6.0. Ergänzende trainierbare Modelle: Die CI bleibt grün, Funktionen kommen bei Gelegenheit hinzu.
Upstream
ConvNeXt V2 von Meta AI, MIT code; CC-BY-NC-4.0 weights. Publikation, Quelle
Lizenzen
Code MIT, Gewichte MIT code; CC-BY-NC-4.0 weights. Kommerzielle Nutzung

Install

bash
pip install "libreyolo"

Predict

Python
from libreyolo import LibreYOLO, SAMPLE_IMAGE model = LibreYOLO("LibreConvNeXtV2atto-cls.pt", device="cpu")result = model(SAMPLE_IMAGE)print(result.probs)

The eight sizes run classification at 224 pixels. Official pretrained weights are CC-BY-NC-4.0.

Train

Training rebuilds the classification head for an ImageFolder dataset. cls_pw accepts values from 0 to 1 and defaults to 0. class_weights=True selects the alternative sample-normalized weighting; it cannot combine with cls_pw>0. Weighting settings must match when resuming.

Dataset setup describes the required training data.

Python
from libreyolo import LibreYOLO model = LibreYOLO("LibreConvNeXtV2atto-cls.pt", device="cpu")# "smoke10" auto-downloads; for your own data, pass a folder with train/ and val/ class subfolders.model.train(data="smoke10", epochs=1, device="cpu", workers=0)

Validate

Python
from libreyolo import LibreYOLO model = LibreYOLO("LibreConvNeXtV2atto-cls.pt", device="cpu")metrics = model.val(data="smoke10", workers=0)print(metrics)

Use the dataset format for this task. Validation explains the dataset requirements and returned metrics.

Export

AufgabeONNXTorchScriptExecuTorchTensorRTOpenVINOPaddleMNNRKNNncnnTFLiteCoreMLCore AI
ClassificationClassification to ONNX: unterstütztClassification to TorchScript: unterstütztClassification to ExecuTorch: unterstütztClassification to TensorRT: unterstütztClassification to OpenVINO: unterstütztClassification to Paddle: nicht unterstütztClassification to MNN: nicht unterstütztClassification to RKNN: nicht unterstütztClassification to ncnn: unterstütztClassification to TFLite: nicht unterstütztClassification to CoreML: nicht unterstütztClassification to Core AI: nicht unterstützt
Python
from libreyolo import LibreYOLO, SAMPLE_IMAGE model = LibreYOLO("LibreConvNeXtV2atto-cls.pt", device="cpu")model.export(format="onnx")

Export setup lists format dependencies and loading exported artifacts.

Checkpoints

DateiLizenz der Gewichte
Classification
LibreConvNeXtV2atto-cls.ptcc-by-nc-4.0
LibreConvNeXtV2femto-cls.ptcc-by-nc-4.0
LibreConvNeXtV2pico-cls.ptcc-by-nc-4.0
LibreConvNeXtV2n-cls.ptcc-by-nc-4.0
LibreConvNeXtV2t-cls.ptcc-by-nc-4.0
LibreConvNeXtV2b-cls.ptcc-by-nc-4.0
LibreConvNeXtV2l-cls.ptcc-by-nc-4.0
LibreConvNeXtV2h-cls.ptcc-by-nc-4.0

Jede oben aufgeführte Datei ist heute in der LibreYOLO-Organisation verfügbar und wird bei der ersten Verwendung heruntergeladen.

Licensing

Prüfe die Lizenz im Hugging-Face-Repository der konkreten Gewichte, die du herunterlädst. Jeder Checkpoint in der LibreYOLO-Organisation hat eine Lizenz, und sie ist innerhalb einer Familie nicht immer gleich. Dieses Repository ist die maßgebliche Quelle. Die Zusammenfassung unten beschreibt den Stand bei der letzten Verifizierung dieser Seite.

Dies ist eine Beschreibung der beteiligten Lizenzen und keine Rechtsberatung. Wenn die Antwort kommerziell relevant ist, lies die Lizenzen selbst und hole eigenen Rechtsrat ein.

Originalarbeit
ConvNeXt V2, Meta AI
Upstream-Lizenz
MIT code; CC-BY-NC-4.0 weights
LibreYOLO-Code
MIT
Gewichte
MIT code; CC-BY-NC-4.0 weights, erneut unter huggingface.co/LibreYOLO veröffentlicht
Einordnung
The source code is MIT. Official pretrained weights are CC-BY-NC-4.0 and exclude commercial use.

Citation

@article{Woo2023ConvNeXtV2,
  title={ConvNeXt V2: Co-designing and Scaling ConvNets with Masked Autoencoders},
  author={Sanghyun Woo, Shoubhik Debnath, Ronghang Hu, Xinlei Chen, Zhuang Liu, In So Kweon and Saining Xie},
  year={2023},
  journal={arXiv preprint arXiv:2301.00808},
}

Aus dem Zitierblock der Autoren unter raw.githubusercontent.com/facebookresearch/ConvNeXt-V2/main/README.md kopiert.

Mit LibreYOLO v1.6.0 verifiziert. Supporttabellen, Checkpoints und Benchmarkwerte auf dieser Seite werden aus der veröffentlichten Bibliothek und den publizierten Gewichten generiert und nicht von Hand geschrieben.