lightly-ai/lightly
A python library for self-supervised learning on images.
解決的問題
自我監督學習 (SSL) 使模型能夠從無標籤數據中學習，從而減少對人工數據標註的沉重依賴。此框架提供了實現複雜 SSL 演算法所需的構建塊，無需從零開始。
工作原理
Lightly 提供了一個模組化框架，公開了損失函數和模型頭等底層組件。它設計用於與 PyTorch 配合使用，並透過 PyTorch Lightning 支援分散式訓練。它允許使用者透過類 PyTorch 介面實驗各種骨幹網路並實現最先進的 SSL 方法。
適用對象
需要使用先進自我監督學習技術在無標籤數據集上進行模型預訓練的電腦視覺研究人員和機器學習工程師。