YOLO-NAS Pose:用於訓練PyTorch計算機視覺模型的開源庫

Link:yolo-nas-pose
YOLO-NAS Pose是一款免費的、開源的庫,用於訓練基於PyTorch的計算機視覺模型。它提供了訓練指令碼和快速簡單複製模型結果的示例。內建SOTA模型,可以輕鬆載入和微調生產就緒的預訓練模型,包括最佳實踐和驗證的超引數,以實現最佳的準確性。可以縮短訓練生命週期,消除不確定性。提供分類、檢測、分割等不同任務的模型,可以輕鬆整合到程式碼庫中。

需求人群:

"用於訓練和微調計算機視覺模型,可以用於各種計算機視覺任務。"

使用場景示例:

用YOLO-NAS Pose訓練一個影象分類模型

使用YOLO-NAS Pose微調一個目標檢測模型

利用YOLO-NAS Pose訓練一個影象分割模型

產品特色:

提供訓練指令碼和模型結果的快速簡單複製示例

內建SOTA模型,可以輕鬆載入和微調生產就緒的預訓練模型

提供分類、檢測、分割等不同任務的模型

返回頂端