Project Title

pytorch-image-models — The Largest Collection of Pretrained PyTorch Image Models

Overview

PyTorch Image Models is a comprehensive library offering a vast array of PyTorch image encoders and backbones, including ResNet, ResNeXT, EfficientNet, and Vision Transformer (ViT), among others. It stands out for its extensive collection of models, support for training, evaluation, inference, and exporting scripts, as well as providing pretrained weights. This project is a one-stop solution for developers needing a variety of image classification models in PyTorch.

Key Features

Over 30 different state-of-the-art image models available
Pretrained weights for immediate use in projects
Support for training, evaluation, inference, and exporting models
Regular updates and additions of new models and features

Use Cases

Researchers and developers using deep learning for image classification tasks
Applications requiring various image encoders for different levels of complexity and performance
Rapid prototyping and deployment of image recognition systems

Advantages

Extensive model selection covering a wide range of architectures
Easy integration with PyTorch, facilitating the use of these models in existing projects
Community-driven with regular updates and improvements

Limitations / Considerations

The project's large scope may lead to a steeper learning curve for new users
The performance of each model can vary significantly based on the specific use case and dataset

TensorFlow Model Garden: A collection of TensorFlow models, similar in scope but for the TensorFlow framework.
Torchvision: A package that provides pre-trained models and common datasets for computer vision tasks in PyTorch, but with a smaller selection of models compared to pytorch-image-models.

Basic Information

GitHub: https://github.com/huggingface/pytorch-image-models
Stars: 35,073
License: Unknown
Last Commit: 2025-08-20

📊 Project Information

Project Name: pytorch-image-models
GitHub URL: https://github.com/huggingface/pytorch-image-models
Programming Language: Python
⭐ Stars: 35,073
🍴 Forks: 4,996
📅 Created: 2019-02-02
🔄 Last Updated: 2025-08-20

🏷️ Project Topics

Topics: [, ", a, u, g, m, i, x, ", ,, , ", c, o, n, v, n, e, x, t, ", ,, , ", d, i, s, t, r, i, b, u, t, e, d, -, t, r, a, i, n, i, n, g, ", ,, , ", e, f, f, i, c, i, e, n, t, n, e, t, ", ,, , ", i, m, a, g, e, -, c, l, a, s, s, i, f, i, c, a, t, i, o, n, ", ,, , ", i, m, a, g, e, n, e, t, ", ,, , ", m, a, x, v, i, t, ", ,, , ", m, i, x, n, e, t, ", ,, , ", m, o, b, i, l, e, -, d, e, e, p, -, l, e, a, r, n, i, n, g, ", ,, , ", m, o, b, i, l, e, n, e, t, -, v, 2, ", ,, , ", m, o, b, i, l, e, n, e, t, v, 3, ", ,, , ", n, f, n, e, t, s, ", ,, , ", n, o, r, m, a, l, i, z, a, t, i, o, n, -, f, r, e, e, -, t, r, a, i, n, i, n, g, ", ,, , ", o, p, t, i, m, i, z, e, r, ", ,, , ", p, r, e, t, r, a, i, n, e, d, -, m, o, d, e, l, s, ", ,, , ", p, r, e, t, r, a, i, n, e, d, -, w, e, i, g, h, t, s, ", ,, , ", p, y, t, o, r, c, h, ", ,, , ", r, a, n, d, a, u, g, m, e, n, t, ", ,, , ", r, e, s, n, e, t, ", ,, , ", v, i, s, i, o, n, -, t, r, a, n, s, f, o, r, m, e, r, -, m, o, d, e, l, s, ", ]

📚 Documentation

Getting Started (Documentation)
Gemma 3n

What's New
Introduction
Models
Features
Results

This article is automatically generated by AI based on GitHub project information and README content analysis

pytorch-image-models

Project Description