https://github.com/veb-101/keras-vision

Porting vision models to Keras 3 for easily accessibility. Contains MobileViT v1, MobileViT v2
https://github.com/veb-101/keras-vision

cnn computer-vision deep-learning jax keras keras3 mobilevit mobilevitv1 mobilevitv2 pytorch self-attention tensorflow transformers

Last synced: 3 months ago
JSON representation

Porting vision models to Keras 3 for easily accessibility. Contains MobileViT v1, MobileViT v2

• Host: GitHub
• URL: https://github.com/veb-101/keras-vision
• Owner: veb-101
• Created: 2024-04-07T16:04:51.000Z (10 months ago)
• Default Branch: main
• Last Pushed: 2024-10-15T20:38:09.000Z (3 months ago)
• Last Synced: 2024-10-31T19:23:41.441Z (3 months ago)
• Topics: cnn, computer-vision, deep-learning, jax, keras, keras3, mobilevit, mobilevitv1, mobilevitv2, pytorch, self-attention, tensorflow, transformers
• Language: Jupyter Notebook
• Homepage:
• Size: 2.35 MB
• Stars: 8
• Watchers: 3
• Forks: 0
• Open Issues: 1
• Metadata Files:
  • Readme: Readme.md

Awesome Lists containing this project

README

        
[![PyPI - Python Version](https://img.shields.io/pypi/pyversions/keras-vision)](https://www.python.org/)  [![PyPI version](https://badge.fury.io/py/keras-vision.svg)](https://badge.fury.io/py/keras-vision) [![Keras](https://img.shields.io/badge/Keras%203.x-%23D00000.svg?logo=Keras&logoColor=white)](https://github.com/keras-team/keras/releases) ![PyPI - Downloads](https://img.shields.io/pypi/dm/keras-vision?style=plastic&logo=Keras&logoColor=red&link=https%3A%2F%2Fpypi.org%2Fproject%2Fkeras-vision%2F)

Porting all models from everywhere to Keras to leverage multi-backend support.

Cause why not?🤷🏻‍♂️

# Table of Contents

- [Table of Contents](#table-of-contents)

  - [Progress](#progress)

  - [Updates](#updates)

  - [Quick Setup](#quick-setup)

    - [Stable PyPi Package](#stable-pypi-package)

    - [Latest Git Updates](#latest-git-updates)

  - [Models Supported](#models-supported)

## Progress

- Working on adding MobileNet V4

## Updates

1. [2024-06-24] Released MobileViT v2 - All Image Classification variants.

2. [2024-05-15] Fixed MobileViT v1 - Now works will all 3 backends. 🎉🎉

3. [2024-05-04] Converted MobileViT to Keras 3 and released weights of all 3 variants.

   1. Jax backend currently not working, I'm working on a fix.

   2. Release: 

## Quick Setup

### Stable PyPi Package

```bash

pip install -U keras-vision

```

### OR

### Latest Git Updates

```bash

pip install git+https://github.com/veb-101/keras-vision.git

```

## Models Supported

   

      

         

            # 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