lightly-ai/lightly
A python library for self-supervised learning on images.
解决的问题
自监督学习 (SSL) 使模型能够从无标签数据中学习，从而减少对人工数据标注的沉重依赖。该框架提供了实现复杂 SSL 算法所需的构建块，无需从零开始。
工作原理
Lightly 提供了一个模块化框架，公开了损失函数和模型头等底层组件。它设计用于与 PyTorch 配合使用，并通过 PyTorch Lightning 支持分布式训练。它允许用户使用类 PyTorch 接口实验各种骨干网络并实现最先进的 SSL 方法。
适用对象
需要使用先进自监督学习技术在无标签数据集上进行模型预训练的计算机视觉研究人员和机器学习工程师。