lightly-ai/lightly

解決的問題

LightlySSL 是一個為使自監督學習（SSL）更易存取且更易實作而設計的電腦視覺框架。它提供模組化系統，可在未標記資料上預訓練模型，減少電腦視覺任務對大型手動標記資料集的依賴。

工作原理

此框架以類似 PyTorch 的模組化程式庫形式建構，公開低階的建構模組，例如損失函數與模型頭。使用者可將不同主幹模型與各種 SSL 算法組合使用。同時支援透過 PyTorch Lightning 進行分散式訓練，以處理更大規模的資料集與運算資源。

適用對象

適合希望實作與實驗最新自監督學習方法，但無需從零開始建構一切的電腦視覺研究人員與開發者。

主要特色

• 廣泛的模型支援：包含 MoCo、SimCLR、BYOL、DINO、MAE 和 LeJEPA 等流行 SSL 模型的實作。
• 模組化設計：提供靈活的損失函數與模型頭建構模組。
• 客製化：支援自訂主幹模型用於預訓練。
• 整合性：完全相容 PyTorch 與 PyTorch Lightning，支援可擴展訓練。