#446 · Primary category: Computer Vision

CvT

classification computer-vision cvt deep-learning imagenet

This is an official implementation of CvT: Introducing Convolutions to Vision Transformers.

Project last updated:05/16/23

GitHub Stars

606

Forks

129

Contributors

5

License

MIT

Why we included this project

CvT is Microsoft's official implementation of a vision transformer that folds convolutions into the standard ViT design, picking up the shift, scale, and distortion invariance CNNs are known for while keeping the global attention that makes transformers work. The paper's headline result is that this hybrid beats plain ViTs and ResNets on ImageNet-1k while using fewer parameters and FLOPs, and the repo ships the exact training and evaluation code behind those numbers, organized around YAML configs and a run script. Pretrained checkpoints for the CvT-13, CvT-21, and CvT-W24 variants are included, so you can reproduce the published results or fine-tune a model to your own dataset without reimplementing anything. It is a research codebase rather than a pip-installable library, so expect to work directly in PyTorch and adapt the training loop to your setup. If you are comparing efficient vision backbones or want a solid starting point for classification experiments, the pretrained weights and clean experiment structure make it worth a look.

Articles for this project

No articles for this project yet.

To suggest a topic or contribute an article, contact us.

Related projects in this category