#345 · Primary category: Computer Vision

PaddleViT

classification computer-vision cv deep-learning detection encoder-decoder gan mlp object-detection paddlepaddle segmentation semantic-segmentation transformer vit

:robot: PaddleViT: State-of-the-art Visual Transformer and MLP Models for PaddlePaddle 2.0+

Project last updated:09/07/22

GitHub Stars

1.2K

Forks

328

Contributors

28

License

Apache-2.0

Why we included this project

PaddleViT is a practical model collection for teams building computer vision systems on PaddlePaddle. It packages working implementations of well-known transformer and MLP architectures like ViT, DeiT, Swin, and BEiT, along with pretrained weights you can fine-tune on your own data instead of training from scratch. Each architecture is a standalone Python module, so you can pull out a single model for a quick experiment without importing the whole codebase. The same repo also covers object detection, semantic segmentation, and GANs, and bundles the training and validation scripts, mixed-precision and multi-GPU support, and export tooling that help when a model needs to move from research into a deployed service. For researchers and engineers who build on PaddlePaddle, it saves the trouble of reimplementing current vision models by hand.

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