/MobileViT-Implementation

MobileViT Implementation from scratch in TensorFlow and PyTorch

Primary LanguagePython

MobileViT

MobileViT is a light-weight and general-purpose vision transformer for mobile vision tasks. It combines the strength of the standard CNN and the ViT. It has outperforms several CNNs and ViT-based network across different tasks and datasets.

Architecture

The block diagram of the MobileViT along with the TMobileViT block
The block diagram of the MobileViT along with the TMobileViT block.

Contact:

For more follow me on: