MiT (SegFormer encoder)
Collection
Pure-Keras 3 MiT (SegFormer encoder) weights (classification and segmentation) for kerasformers. • 6 items • Updated
How to use kerasformers/mit_b5_in1k with KerasFormers:
# No code snippets available yet for this library. # To use this model, check the repository files and the library's documentation. # Want to help? PRs adding snippets are welcome at: # https://github.com/huggingface/huggingface.js
How to use kerasformers/mit_b5_in1k with Keras:
# Available backend options are: "jax", "torch", "tensorflow".
import os
os.environ["KERAS_BACKEND"] = "jax"
import keras
model = keras.saving.load_model("hf://kerasformers/mit_b5_in1k")
Pure-Keras 3 image-classification weight for kerasformers, converted from nvidia/mit-b5.
from kerasformers.models.mit import MiTImageClassify
model = MiTImageClassify.from_weights("mit_b5_in1k")
License: other (nvidia-segformer-license), inherited from the upstream source.
Base model
nvidia/mit-b5