← all repositories

SwinTransformer/Transformer-SSL

A PyTorch implementation of self-supervised learning pretraining with Swin Transformer backbones, evaluating learned representations on downstream computer vision tasks.

669 stars Python ML FrameworksComputer Vision
Transformer-SSL
Velocity · 7d
+0.4
★ / day
Trend
steady
star history

This repository provides the official implementation of self-supervised learning using vision transformers, specifically Swin Transformer. The project trains models via self-supervision to learn visual representations that transfer well to downstream tasks including object detection, instance segmentation, and semantic segmentation. It achieves competitive ImageNet-1K linear evaluation accuracy using DeiT-S/16 with 300 epoch pretraining while requiring fewer training tricks than comparable approaches like MoCo v3 and DINO.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.