Super-Selfish：基于 PyTorch 的图像自监督学习框架
计算机视觉与模式识别
2020-12-08 v2
摘要
Super-Selfish 是一个易于使用的 PyTorch 框架，用于基于图像的自监督学习。可通过 13 种算法学习特征，涵盖从简单分类到更复杂的 SOTA 对比 pretext 任务。该框架易于使用，仅需两行代码即可对任意 PyTorch 神经网络进行预训练。同时，通过模块化设计选择保持了充分的灵活性。代码可在 https://github.com/MECLabTUDA/Super_Selfish 获取，并使用 pip install super-selfish 安装。
引用
@article{arxiv.2012.02706,
title = {Super-Selfish: Self-Supervised Learning on Images with PyTorch},
author = {Nicolas Wagner and Anirban Mukhopadhyay},
journal= {arXiv preprint arXiv:2012.02706},
year = {2020}
}
备注
Technical Report