BobMcDear/vit-pytorch
PyTorch implementation of the vision transformer
This project helps machine learning engineers or researchers implement a Vision Transformer (ViT) architecture in PyTorch. It takes raw image data as input and outputs classifications based on the visual content. The end-user is a deep learning practitioner working on computer vision tasks.
No commits in the last 6 months.
Use this if you are a machine learning engineer or researcher building custom computer vision models and need a flexible, modular PyTorch implementation of the Vision Transformer.
Not ideal if you are looking for a pre-trained model or a high-level API for immediate image classification without diving into model architecture details.
Stars
17
Forks
—
Language
Python
License
GPL-3.0
Category
Last pushed
Mar 30, 2023
Commits (30d)
0
Get this data via API
curl "https://pt-edge.onrender.com/api/v1/quality/ml-frameworks/BobMcDear/vit-pytorch"
Open to everyone — 100 requests/day, no key needed. Get a free key for 1,000/day.
Higher-rated alternatives
Jittor/jittor
Jittor is a high-performance deep learning framework based on JIT compiling and meta-operators.
berniwal/swin-transformer-pytorch
Implementation of the Swin Transformer in PyTorch.
zhanghang1989/ResNeSt
ResNeSt: Split-Attention Networks
NVlabs/FasterViT
[ICLR 2024] Official PyTorch implementation of FasterViT: Fast Vision Transformers with...
ViTAE-Transformer/ViTPose
The official repo for [NeurIPS'22] "ViTPose: Simple Vision Transformer Baselines for Human Pose...