Skip to content

Latest commit

 

History

History
 
 

segformer

Folders and files

NameName
Last commit message
Last commit date

parent directory

..
 
 
 
 
 
 

candle-segformer

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