vision-transformer-models topic

List vision-transformer-models repositories

pytorch-image-models

30.2k
Stars
4.6k
Forks
291
Watchers

The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights -- ResNet, ResNeXT, EfficientNet, NFNet, Vision Transformer (ViT)...

pytorch-image-models-with-simclr

35
Stars
7
Forks
Watchers

Multi-label classification based on timm, and add SimCLR to timm.