lightly

收录理由

对图像数据量大、标注成本高的团队来说，Lightly 提供了一条务实路径：无需人工标注即可预训练编码器，并把它作为下游任务的起点复用。库内实现了 SimCLR、BYOL、SwAV、MoCo、SimSiam、Barlow Twins 等多种成熟的自监督方法，API 风格贴近 PyTorch，模块化设计让损失函数、模型头等组件可以灵活拆装，想要实验自定义骨干网络时，不必与僵化的整体流程纠缠。项目还附带 PyTorch Lightning 示例，并支持分布式训练，从研究原型过渡到多 GPU 环境会更顺畅。如果你曾希望在花钱请人标注之前，先为图像数据搭起一个可用的特征表示，这个库值得一试。