← All repos

huggingface/pytorch-image-models

View on GitHub ↗ ★ 37,273 Python

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

4 mentions on Hacker News by 4 people: 1 story, 3 comments · first 2023-04-10 · latest 2024-11-19