多码网
返回 AI
AIlibraryPython

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

36,680 stars5,146 forks66 issuesApache-2.0更新于 2026/4/22
augmixconvnextdistributed-trainingefficientnetimage-classificationimagenetmaxvitmixnetmobile-deep-learningmobilenet-v2mobilenetv3nfnetsnormalization-free-trainingoptimizerpretrained-modelspretrained-weightspytorchrandaugmentresnetvision-transformer-models
暂无详细内容

同类推荐