MobileViTv2 (mobilevitv2 1.0 imagenet1k 256) MobileViTv2 is the second version of MobileViT. It was proposed in Separable Self attention for Mobile Vision Transformers by Sachin Mehta and Mohammad Rastegari, and first released in this repository. The license used is Apple sample code license. Disclaimer: The team releasing MobileViT did not write a model card for this model so this model card has been written by the Hugging Face team. Model Description MobileViTv2 is constructed by replacing the multi headed self attention in MobileViT with separable self attention. Intended uses & limitations You can use the raw model for image classification. See the model hub to look for fine tuned versions on a task that interests you. How to use Here is how to use this model to classify an image of the COCO 2017 dataset into one of the 1,000 ImageNet classes: Currently, both the feature extractor and model support PyTorch. Training data The MobileViT model was pretrained on ImageNet 1k, a dataset consisting of 1 million images and 1,000 classes. BibTeX entry and citation info
We use cookies for essential functionality and analytics. You can accept or reject analytics cookies.Cookie policy