YOLO-NAS 仍在维护
Xuban
YOLO-NAS 仍然实用。它原来所在的 SuperGradients 最近一次发版是 2024 年 4 月的 3.7.1。LibreYOLO 为它提供预测、训练、验证和导出。
from libreyolo import LibreYOLO
model = LibreYOLO("LibreYOLONASs.pt")
result = model("image.jpg", save=True)
print(result.boxes.xyxy)
LibreYOLO 支持检测的 S、M、L 三档，以及姿态估计。检查点从 Deci 的公开 CDN 下载，文件名形如 LibreYOLONASs.pt。LibreYOLO 并不托管这些权重，Deci 的非商业条款仍然适用。SuperGradients 把 PyTorch 钉在 1.9 到 1.13；LibreYOLO 要求 2.4 或更高。检测导出覆盖 ONNX、TorchScript、OpenVINO、NCNN、TFLite、Paddle、MNN、ExecuTorch 和 Core AI。不支持 CoreML。
若要在没有 Deci 检查点的情况下训练：
from libreyolo import LibreYOLONAS
model = LibreYOLONAS(None, size="s")
model.train(data="my-dataset.yaml", imgsz=640, batch=16)
我们也计划在 COCO 上从零训练 YOLO-NAS 并发布这些权重。
运行 pip install libreyolo 即可安装。模型详情见 YOLO-NAS 文档。另有一篇更长的 SuperGradients 替代方案 文章。