veb-101/apple-fastvit-Keras-3

Please check out my github repo: https://github.com/veb-101/keras-vision for more models

Quick Setup

Stable PyPi Package

pip install -U keras-vision

OR

Latest Git Updates

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

import os

os.environ["KERAS_BACKEND"] = "jax"
# os.environ["KERAS_BACKEND"] = "tensorflow"
# os.environ["KERAS_BACKEND"] = "torch"

from keras_vision.fastvit import fastvit_ma36

model = fastvit_ma36(
    pretrained=True,
    inference_mode=True,
    load_kd_weights=True,
)
model.summary()

Weight file suffix clarification:

• keras_fastvit_ma36.weights.h5 - Weights for model variant - ma36
• keras_fastvit_ma36_reparam.weights.h5 - Reparameterized weights for model variant - ma36 (inference_mode = True)
• keras_fastvit_ma36_kd.weights.h5 - Weights for model variant - ma36, trained using knowledge distillation.
• keras_fastvit_ma36_reparam_kd.weights.h5 - Reparameterized weights for model variant - ma36, trained using knowledge distillation (inference_mode = True)