candle-segformer

HuggingFace Segformer Model Card
mit-b0 - An encoder only pretrained model
segformer-b0-finetuned-ade-512-512 - A fine tuned model for segmentation

How to run the example

If you want you can use the example images from this pull request, download them and supply the path to the image as an argument to the example.

bash

# run the image classification task
cargo run --example segformer classify candle-examples/examples/yolo-v8/assets/bike.jpg

# run the segmentation task
cargo run --example segformer segment candle-examples/examples/yolo-v8/assets/bike.jpg

Example output for classification:

text

classification logits [3.275261e-5, 0.0008562019, 0.0008868563, 0.9977506, 0.0002465068, 0.0002241473, 2.846596e-6]
label: hamburger