Porting all models from everywhere to Keras to leverage multi-backend support.
Cause why not?🤷🏻♂️
- Working on adding MobileNet V4
- [2024-06-24] Released MobileViT v2 - All Image Classification variants.
- [2024-05-15] Fixed MobileViT v1 - Now works will all 3 backends. 🎉🎉
- [2024-05-04] Converted MobileViT to Keras 3 and released weights of all 3 variants.
- Jax backend currently not working, I'm working on a fix.
- Release: https://github.com/veb-101/keras-vision/releases/tag/v0.4
pip install -U keras-vision
pip install git+https://github.com/veb-101/keras-vision.git
# No. | Models | Paper | Additional Materials | Example notebook | Weights URL |
---|---|---|---|---|---|
1 | MobileViT-V1 | MobileViT: Light-weight, General-purpose, and Mobile-friendly Vision Transformer | Blogpost: Building MobileViT In Keras 3 | Colab link | Releases v0.4 |
2 | MobileViT-V2 | Separable Self-attention for Mobile Vision Transformers | -- | Colab link | Releases v0.5 |