YOLO-NAS vẫn được duy trì: dự đoán, huấn luyện và xuất với LibreYOLO
YOLO-NAS vẫn là một mô hình phát hiện hữu ích. Nơi phát triển ban đầu của mô hình là SuperGradients, dự án này phát hành phiên bản 3.7.1 lần cuối vào tháng 4 năm 2024. LibreYOLO duy trì các tính năng dự đoán, huấn luyện, kiểm định và xuất cho mô hình này.
from libreyolo import LibreYOLO
model = LibreYOLO("LibreYOLONASs.pt")
result = model("image.jpg", save=True)
print(result.boxes.xyxy)
LibreYOLO hỗ trợ các cỡ S, M và L cho tác vụ phát hiện, đồng thời hỗ trợ cả tác vụ pose. Các checkpoint được tải từ CDN công khai của Deci với tên như LibreYOLONASs.pt. LibreYOLO không lưu trữ các checkpoint này, và các điều khoản phi thương mại của Deci vẫn được áp dụng. SuperGradients ghim phiên bản PyTorch trong khoảng từ 1.9 đến 1.13; LibreYOLO yêu cầu phiên bản 2.4 trở lên. Tính năng xuất cho phát hiện hỗ trợ ONNX, TorchScript, OpenVINO, NCNN, TFLite, Paddle, MNN, ExecuTorch và Core AI. CoreML không được hỗ trợ.
Để huấn luyện mà không dùng checkpoint của Deci:
from libreyolo import LibreYOLONAS
model = LibreYOLONAS(None, size="s")
model.train(data="my-dataset.yaml", imgsz=640, batch=16)
Chúng tôi cũng dự định huấn luyện YOLO-NAS từ đầu trên COCO và phát hành các trọng số đó.
Cài đặt bằng lệnh pip install libreyolo. Tài liệu YOLO-NAS có thông tin chi tiết về mô hình. Bài viết Lựa chọn thay thế SuperGradients cung cấp phần giải thích dài hơn.