YOLO-NASはまだ使える？LibreYOLOで学習・推論・エクスポート
Xuban
YOLO-NASは今も有用な検出モデルです。元の提供元であるSuperGradientsの最終リリースは、2024年4月の3.7.1でした。LibreYOLOでは、YOLO-NASの推論、学習、検証、エクスポートをメンテナンスしています。
from libreyolo import LibreYOLO
model = LibreYOLO("LibreYOLONASs.pt")
result = model("image.jpg", save=True)
print(result.boxes.xyxy)
LibreYOLOは検出タスクでS、M、Lをサポートし、姿勢推定にも対応しています。チェックポイントはLibreYOLONASs.ptのような名前でDeciの公開CDNからダウンロードされます。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)
YOLO-NASをCOCOでスクラッチから学習し、その重みを公開する予定もあります。
pip install libreyoloでインストールできます。YOLO-NASのドキュメントにモデルの詳細があります。SuperGradientsの代替に関する、より詳しい記事もあります。