mirror of
https://github.com/huggingface/candle.git
synced 2025-06-15 10:26:33 +00:00
924ccae30c
* add segformer * Make the id2label field optional. --------- Co-authored-by: laurent <laurent.mazare@gmail.com>
1011 B
1011 B
candle-segformer
- HuggingFace Segformer Model Card
mit-b0- An encoder only pretrained modelsegformer-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.
# run the image classification task
cargo run --example segformer classify <path-to-image>
# run the segmentation task
cargo run --example segformer segment <path-to-image>
Example output for classification:
classification logits [3.275261e-5, 0.0008562019, 0.0008868563, 0.9977506, 0.0002465068, 0.0002241473, 2.846596e-6]
label: hamburger