keras-vision

Building Vision models in Keras3 for framework-agnostic training and inference.


Keywords
keras3, tensorflow, Jax, PyTorch, Vision, cnn, computer-vision, deep-learning, keras, mobilevit, self-attention, transformers
License
Apache-2.0
Install
pip install keras-vision==0.4.5

Documentation

PyPI - Python Version PyPI version Keras

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

Cause why not?🤷🏻‍♂️

Installation:

  1. Option 1: Install from PyPI: pip install -U keras-vision

  2. Option 2: Install the latest updates present in the repository: pip install git+https://github.com/veb-101/keras-vision.git

Usage: Please refer to the examples folder

Updates

  1. [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: https://github.com/veb-101/keras-vision/releases/tag/v0.4