huggingface/pytorch-image-models

PyTorch画像エンコーダ/バックボーンの最大コレクション。学習、評価、推論、エクスポートスクリプト、事前学習済みウェイトを含む — ResNet, ResNeXT, EfficientNet, NFNet, Vision Transformer (ViT), MobileNetV4, MobileNet-V3 & V2, RegNet, DPN, CSPNet, Swin Transformer, MaxViT, CoAtNet, ConvNeXtなど。