huggingface/pytorch-image-models
The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights -- ResNet, ResNeXT, EfficientNet, NFNet, Vision Transformer (ViT), MobileNetV4, MobileNet-V3 & V2, RegNet, DPN, CSPNet, Swin Transformer, MaxViT, CoAtNet, ConvNeXt, and more
GitHub repository with 36,846 stars and 5,162 forks.
Language: Python
Topics: pytorch, resnet, pretrained-models, pretrained-weights, distributed-training, mobile-deep-learning, mobilenet-v2, mobilenetv3, efficientnet, augmix