open-mmlab/mmselfsup
A PyTorch-based toolbox and benchmark for self-supervised learning, implementing methods like BEiT, MAE, MoCo, SimCLR, and SimSiam for visual model pre-training.

MMSelfSup is a comprehensive open-source toolbox for self-supervised learning in computer vision. It provides implementations of major SSL algorithms including masked image modeling approaches (BEiT, MAE) and contrastive methods (MoCo, SimCLR, SimSiam). The project serves as both a research benchmark and a practical training framework, offering model zoo with pre-trained weights and unified training pipelines built on the OpenMMLab ecosystem.