YOLO-NAS Masih Dipelihara dan Bisa Digunakan dengan LibreYOLO
YOLO-NAS masih merupakan model deteksi yang berguna. Repositori asalnya, SuperGradients, terakhir merilis versi 3.7.1 pada April 2024. LibreYOLO memelihara dukungan prediksi, pelatihan, validasi, dan ekspor untuk model ini.
from libreyolo import LibreYOLO
model = LibreYOLO("LibreYOLONASs.pt")
result = model("image.jpg", save=True)
print(result.boxes.xyxy)
LibreYOLO mendukung ukuran S, M, dan L untuk deteksi, serta pose. Checkpoint diunduh dari CDN publik Deci dengan nama seperti LibreYOLONASs.pt. LibreYOLO tidak meng-host satu pun checkpoint tersebut, dan ketentuan nonkomersial Deci tetap berlaku. SuperGradients membatasi versi PyTorch pada rentang 1.9 hingga 1.13, sedangkan LibreYOLO memerlukan versi 2.4 atau lebih baru. Ekspor deteksi mendukung ONNX, TorchScript, OpenVINO, NCNN, TFLite, Paddle, MNN, ExecuTorch, dan Core AI. CoreML tidak didukung.
Untuk melatih tanpa checkpoint Deci:
from libreyolo import LibreYOLONAS
model = LibreYOLONAS(None, size="s")
model.train(data="my-dataset.yaml", imgsz=640, batch=16)
Kami juga berencana melatih YOLO-NAS dari nol dengan COCO dan menerbitkan bobotnya.
Instal dengan pip install libreyolo. Dokumentasi YOLO-NAS memuat detail model. Baca juga artikel Alternatif SuperGradients yang lebih lengkap.