YOLO-NAS est-il encore maintenu ? Utiliser et entraîner YOLO-NAS avec LibreYOLO
YOLO-NAS reste un détecteur utile. Son projet d'origine, SuperGradients, a publié sa dernière version, la 3.7.1, en avril 2024. LibreYOLO assure la maintenance de la prédiction, de l'entraînement, de la validation et de l'export pour ce modèle.
from libreyolo import LibreYOLO
model = LibreYOLO("LibreYOLONASs.pt")
result = model("image.jpg", save=True)
print(result.boxes.xyxy)
LibreYOLO prend en charge les tailles S, M et L pour la détection, ainsi que l'estimation de pose. Les checkpoints sont téléchargés depuis le CDN public de Deci sous des noms tels que LibreYOLONASs.pt. LibreYOLO n'en héberge aucun, et les conditions non commerciales de Deci restent applicables. SuperGradients impose une version de PyTorch comprise entre 1.9 et 1.13 ; LibreYOLO requiert la version 2.4 ou ultérieure. L'export pour la détection prend en charge ONNX, TorchScript, OpenVINO, NCNN, TFLite, Paddle, MNN, ExecuTorch et Core AI. CoreML n'est pas pris en charge.
Pour entraîner le modèle sans checkpoint de Deci :
from libreyolo import LibreYOLONAS
model = LibreYOLONAS(None, size="s")
model.train(data="my-dataset.yaml", imgsz=640, batch=16)
Nous prévoyons également d'entraîner YOLO-NAS à partir de zéro sur COCO et de publier ces poids.
Installez le package avec pip install libreyolo. La documentation de YOLO-NAS présente les détails du modèle. Consultez aussi notre article plus long sur les alternatives à SuperGradients.