rentainhe/ViT.pytorch
The Pytorch reimplementation of Vision Transformer
This project helps researchers and engineers build and train advanced image recognition models. You provide a dataset of images, and it outputs a trained model capable of classifying those images with high accuracy. It's designed for machine learning practitioners focused on computer vision tasks.
No commits in the last 6 months.
Use this if you need to train a Vision Transformer model for image classification or fine-tune a pre-trained model on your custom image dataset.
Not ideal if you are looking for a plug-and-play image classification solution without needing to engage in model training or customization.
Stars
10
Forks
—
Language
Jupyter Notebook
License
MIT
Category
Last pushed
Oct 01, 2021
Commits (30d)
0
Get this data via API
curl "https://pt-edge.onrender.com/api/v1/quality/ml-frameworks/rentainhe/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...