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

语言：Python · GitHub 星标：37,261 · Fork：5,223

话题：augmix · convnext · distributed-training · efficientnet · image-classification · imagenet · maxvit · mixnet

在 GitHub 上查看原仓库