godofpdog/ViT_PyTorch
This is a simple PyTorch implementation of Vision Transformer (ViT) described in the paper "An Image is Worth 16x16 Words: Transformers for Image Recognition at Scale"
This project helps machine learning engineers and researchers quickly set up and train a Vision Transformer (ViT) model for image classification tasks. You input a dataset of images, and it outputs a trained model capable of categorizing new images. This is for professionals building advanced computer vision systems.
No commits in the last 6 months.
Use this if you need a straightforward PyTorch implementation of the Vision Transformer architecture to classify images.
Not ideal if you are looking for a high-level, no-code solution for image classification or do not have experience with deep learning frameworks like PyTorch.
Stars
25
Forks
3
Language
Python
License
—
Category
Last pushed
Mar 06, 2023
Commits (30d)
0
Get this data via API
curl "https://pt-edge.onrender.com/api/v1/quality/transformers/godofpdog/ViT_PyTorch"
Open to everyone — 100 requests/day, no key needed. Get a free key for 1,000/day.
Compare
Higher-rated alternatives
jaehyunnn/ViTPose_pytorch
An unofficial implementation of ViTPose [Y. Xu et al., 2022]
UdbhavPrasad072300/Transformer-Implementations
Library - Vanilla, ViT, DeiT, BERT, GPT
tintn/vision-transformer-from-scratch
A Simplified PyTorch Implementation of Vision Transformer (ViT)
icon-lab/ResViT
Official Implementation of ResViT: Residual Vision Transformers for Multi-modal Medical Image Synthesis
gupta-abhay/pytorch-vit
An Image is Worth 16x16 Words: Transformers for Image Recognition at Scale