#13 · Primary category: Model Zoos & Hubs
Awesome-Backbones
Integrate deep learning models for image classification | Backbone learning/comparison/magic modification project
Project last updated:01/17/25
GitHub Stars
2.0K
Forks
270
Contributors
5
License
Other
Why we included this project
This project collects PyTorch implementations of a broad set of image classification backbones, from classic LeNet and ResNet to newer EfficientNet, ConvNeXt, Swin Transformer, DeiT, and MobileViT. Each model comes with its own config file, so you can train, evaluate, and swap backbones without rewriting the surrounding pipeline. The tutorials walk through environment setup, dataset preparation, config explanation, and adding new model components, which helps students and engineers understand the internals rather than just calling a pretrained API. It also includes scripts for computing FLOPs and parameters, visualizing class activation maps, and exporting to ONNX, making it a handy comparison and experimentation toolkit. For teams working with small custom datasets, the practical notes on tuning learning rate, batch size, and data augmentation are a real time-saver.
Articles for this project
No articles for this project yet.
To suggest a topic or contribute an article, contact us.
Related projects in this category
models.dev
An open-source database of AI models.
Awesome-CoreML-Models
Largest list of models for Core ML (for iOS 11+)
models
Officially maintained, supported by PaddlePaddle, including CV, NLP, Speech, Rec, TS, big models and so on.
PINTO_model_zoo
A repository for storing models that have been inter-converted between various frameworks. Supported frameworks are TensorFlow, PyTorch, ONNX, OpenVINO, TFJS, TFTRT, TensorFlowLite (Float32/16/INT8), EdgeTPU, CoreML.
huggingface_hub
The official CLI and Python client for the Hugging Face Hub.