segformer

Model type

segformer

Class

SegformerForSemanticSegmentation

Task

image-classification

Source

models/segformer.py

Description

Segformer for semantic segmentation.

Usage

mobius build --model <MODEL_ID> output_dir/
from mobius import build

model = build("<MODEL_ID>")