huggingface/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)
- Stars
- 37k
- Forks
- 5.2k
- Open issues
- 10
- Last push
- 1d ago
About pytorch-image-models
pytorch-image-models is a free, open-source machine learning tool by huggingface, written mainly in 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). It has 37k GitHub stars and 5.2k forks, and was last updated 1d ago. It's released under the Apache-2.0 license, a standard open-source license.
- License
- Apache-2.0
- Language
- Python
- Stars
- 37,230
- Last update
- 1d ago
- Runs on Mac
- Not stated
- GPU
- Not stated
- Paid API key
- Not stated
- Self-hostable
- Not stated
Similar free tools
ludwig12k stars · Low-code framework for building custom LLMs, neural networks, and other AI models
SwanLab4.3k stars · ⚡️SwanLab - an open-source, modern-design AI training tracking and visualization tool. Supports Cloud / Self-hosted use. Integrated with PyTorch / Transformers / verl / LLaMA Factory / ms-swift / Ultr
torchgeo4.2k stars · TorchGeo: datasets, samplers, transforms, and pre-trained models for geospatial data
ModelsGenesis796 stars · [MICCAI 2019 Young Scientist Award] [MedIA Best Paper Award] Models Genesis: self-supervised pre-training for 3D medical images. Learns transferable representations from unlabeled CT and MRI volumes,
uni2ts1.6k stars · Unified Training of Universal Time Series Forecasting Transformers
CLUE4.3k stars · 中文语言理解测评基准 Chinese Language Understanding Evaluation Benchmark: datasets, baselines, pre-trained models, corpus and leaderboard
pytorch-image-models FAQ
Is pytorch-image-models free?
Yes. pytorch-image-models is open source and free to use. It's released under the Apache-2.0 license, a standard open-source license.
Does pytorch-image-models run on a Mac?
Its README doesn't say. Check the installation guide on GitHub before you start.
Does pytorch-image-models need a GPU?
Its README doesn't state hardware needs. AI models generally run much faster with a GPU.
Is pytorch-image-models still maintained?
Yes. It was last updated 1d ago and has 10 open issues on GitHub.
What are alternatives to pytorch-image-models?
Similar free, open-source tools include ludwig, SwanLab, torchgeo, ModelsGenesis.