YOLO-NAS è ancora mantenuto? Usalo con LibreYOLO
YOLO-NAS è ancora un rilevatore utile. La sua piattaforma originale, SuperGradients, ha rilasciato la versione 3.7.1 ad aprile 2024. LibreYOLO mantiene il supporto per la predizione, l'addestramento, la validazione e l'esportazione.
from libreyolo import LibreYOLO
model = LibreYOLO("LibreYOLONASs.pt")
result = model("image.jpg", save=True)
print(result.boxes.xyxy)
LibreYOLO supporta le taglie S, M e L per il rilevamento, oltre alla posa. I checkpoint si scaricano dal CDN pubblico di Deci con nomi come LibreYOLONASs.pt. LibreYOLO non ne ospita nessuno e continuano ad applicarsi le condizioni non commerciali di Deci. SuperGradients vincola PyTorch a versioni dalla 1.9 alla 1.13; LibreYOLO richiede la 2.4 o successive. L'esportazione per il rilevamento supporta ONNX, TorchScript, OpenVINO, NCNN, TFLite, Paddle, MNN, ExecuTorch e Core AI. CoreML non è supportato.
Per addestrare senza un checkpoint di Deci:
from libreyolo import LibreYOLONAS
model = LibreYOLONAS(None, size="s")
model.train(data="my-dataset.yaml", imgsz=640, batch=16)
Puntiamo anche ad addestrare YOLO-NAS da zero su COCO e a pubblicare quei pesi.
Installa con pip install libreyolo. La documentazione di YOLO-NAS descrive il modello. Nell'articolo Un'alternativa a SuperGradients trovi un approfondimento.