#168 · 主分类： 计算机视觉
byol-pytorch
artificial-intelligence
deep-learning
self-supervised-learning
Deepmind的'Bootstrap Your Own Latent'自监督学习的可用实现，基于Pytorch。
项目最后更新：04/27/26
GitHub Stars
1.9K
Forks数量
248
贡献者数量
6
许可证
MIT
收录理由
标注数据不足时，想给视觉模型加一轮预训练，这个库提供了一种轻量的做法。它实现了DeepMind提出的BYOL方法，不需要构造负样本对，靠一个滑动平均的目标编码器就能从无标注图像里学表征。用法很直接：把任意基于图像的网络（比如ResNet）包进去，指定图像尺寸和隐藏层，喂原始图像训练就行。训练结束后，同一个网络在后续有监督任务上通常表现更好。代码量小，用pip就能安装，研究者读起来容易理解，工程师在有限标注集上微调前想快速加一步预训练也很顺手。
相关文章
暂无针对该项目的文章。
如需补充新闻资讯、使用教程、应用测评等内容，欢迎 联系我们推荐选题或投稿。