lucidrains/vit-pytorch
Implementation of Vision Transformer, a simple way to achieve SOTA in vision classification with only a single transformer encoder, in Pytorch
- Stars
- 26k
- Forks
- 3.5k
- Open issues
- 132
- Last push
- 9d ago
About vit-pytorch
vit-pytorch is a free, open-source computer vision tool by lucidrains, written mainly in Python. Implementation of Vision Transformer, a simple way to achieve SOTA in vision classification with only a single transformer encoder, in Pytorch. It has 26k GitHub stars and 3.5k forks, and was last updated 9d ago. It's released under the MIT license, a standard open-source license.
- License
- MIT
- Language
- Python
- Stars
- 25,550
- Last update
- 9d ago
- Runs on Mac
- Not stated
- GPU
- Not stated
- Paid API key
- Not stated
- Self-hostable
- Yes
Similar free tools
ultralytics63k stars · Ultralytics YOLO27, YOLO26, YOLO11, YOLOv8 — object detection, instance segmentation, semantic segmentation, image classification, pose estimation, object tracking
pytorch-grad-cam13k stars · Advanced AI Explainability for computer vision. Support for CNNs, Vision Transformers, Classification, Object detection, Segmentation, Image similarity and more.
cvat17k stars · CVAT is a leading data annotation platform for image, video, audio, and 3D datasets. It offers open-source, cloud, and enterprise products, as well as labeling services, with AI-assisted labeling, qua
label-studio29k stars · Label Studio is a multi-type data labeling and annotation tool with standardized output format
supervision51k stars · We write your reusable computer vision tools. 💜
vit-pytorch FAQ
Is vit-pytorch free?
Yes. vit-pytorch is open source and free to use. It's released under the MIT license, a standard open-source license.
Does vit-pytorch run on a Mac?
Its README doesn't say. Check the installation guide on GitHub before you start.
Does vit-pytorch need a GPU?
Its README doesn't state hardware needs. AI models generally run much faster with a GPU.
Is vit-pytorch still maintained?
Yes. It was last updated 9d ago and has 132 open issues on GitHub.
What are alternatives to vit-pytorch?
Similar free, open-source tools include ultralytics, pytorch-grad-cam, yolov5, cvat.