Downloads · 30 days
22
2% of all-time downloads
mobile-vit-xxs
keras-io/mobile-vit-xxs
mobile-vit-xxs is a image classification model from keras-io. Use it when you need a label for an image. It is set up for tf-keras. The card lists the license as cc0-1.0.
This repo contains the model and the notebook to this Keras example on MobileViT.
All-time downloads
1.4K
Public
Repo size
13.5 MB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.data-00000-of-000015.4 MB · 54%
At a glance
- Task
- Image Classification
- Library
- tf-keras
- License
- cc0-1.0
- Access
- Public
- Created
- Mar 2, 2022
- Updated
- Jul 5, 2024
- SHA
- fa60ea7e
Model card
From the Hugging Face model README
Image Classification using MobileViT
This repo contains the model and the notebook to this Keras example on MobileViT.
Full credits to: Sayak Paul
Background Information
MobileViT architecture (Mehta et al.), combines the benefits of Transformers (Vaswani et al.) and convolutions. With Transformers, we can capture long-range dependencies that result in global representations. With convolutions, we can capture spatial relationships that model locality.
Besides combining the properties of Transformers and convolutions, the authors introduce MobileViT as a general-purpose mobile-friendly backbone for different image recognition tasks. Their findings suggest that, performance-wise, MobileViT is better than other models with the same or higher complexity (MobileNetV3, for example), while being efficient on mobile devices.
Training Data
The model is trained on a tf_flowers dataset
- Task
- Image Classification
- Library
- tf-keras
- License
- cc0-1.0
- Created
- Mar 2, 2022
- Updated
- Jul 5, 2024