YOLO-NAS ainda é mantido: como usar no LibreYOLO
O YOLO-NAS ainda é um detector útil. Seu projeto original, o SuperGradients, lançou a versão 3.7.1 em abril de 2024. O LibreYOLO mantém o suporte a predição, treinamento, validação e exportação para ele.
from libreyolo import LibreYOLO
model = LibreYOLO("LibreYOLONASs.pt")
result = model("image.jpg", save=True)
print(result.boxes.xyxy)
O LibreYOLO oferece suporte aos tamanhos S, M e L para detecção, além de pose. Os checkpoints são baixados do CDN público da Deci com nomes como LibreYOLONASs.pt. O LibreYOLO não hospeda nenhum deles, e os termos não comerciais da Deci continuam valendo. O SuperGradients fixa o PyTorch entre as versões 1.9 e 1.13; o LibreYOLO exige a versão 2.4 ou superior. A exportação para detecção inclui ONNX, TorchScript, OpenVINO, NCNN, TFLite, Paddle, MNN, ExecuTorch e Core AI. CoreML não é compatível.
Para treinar sem um checkpoint da Deci:
from libreyolo import LibreYOLONAS
model = LibreYOLONAS(None, size="s")
model.train(data="my-dataset.yaml", imgsz=640, batch=16)
Também planejamos treinar YOLO-NAS do zero com o COCO e publicar esses pesos.
Instale com pip install libreyolo. A documentação do YOLO-NAS traz os detalhes do modelo. Também temos um artigo mais longo sobre a alternativa ao SuperGradients.