mirror of
https://github.com/huggingface/candle.git
synced 2025-06-17 11:08:52 +00:00
e27aac0a06
* Add: DINOv2Reg4 with PlantCLEF2024 weights and example ( See https://arxiv.org/abs/2309.16588 and https://zenodo.org/records/10848263 ) * Remove extra files + update README to download them + remove extra lines * minor fix (README remove extra spaces) * minor fix (README: Fix image url) * Modif: Add back interpolate_pos_encoding() + fix when no interpolation + remove extra comments + Update README ( source image changed and so the predictions ) * Fix: Improve code lisibility with '$ cargo clippy' and '$ cargo fmt' * Another clippy fix. --------- Co-authored-by: x-VEspit <vincent.espitalier@cirad.fr> Co-authored-by: laurent <laurent.mazare@gmail.com>
26 lines
1.1 KiB
Markdown
26 lines
1.1 KiB
Markdown
# candle-dinov2-reg4
|
||
|
||
[DINOv2-reg4](https://arxiv.org/abs/2309.16588) is the lastest version of DINOv2 with registers.
|
||
In this example, it is used as an plant species classifier: the model returns the
|
||
probability for the image to belong to each of the 7806 PlantCLEF2024 categories.
|
||
|
||
## Running some example
|
||
|
||
```bash
|
||
# Download classes names and a plant picture to identify
|
||
curl https://huggingface.co/vincent-espitalier/dino-v2-reg4-with-plantclef2024-weights/raw/main/species_id_mapping.txt --output candle-examples/examples/dinov2reg4/species_id_mapping.txt
|
||
curl https://bs.plantnet.org/image/o/bd2d3830ac3270218ba82fd24e2290becd01317c --output candle-examples/examples/dinov2reg4/bd2d3830ac3270218ba82fd24e2290becd01317c.jpg
|
||
|
||
# Perform inference
|
||
cargo run --example dinov2reg4 --release -- --image candle-examples/examples/dinov2reg4/bd2d3830ac3270218ba82fd24e2290becd01317c.jpg
|
||
|
||
> Orchis simia Lam. : 45.55%
|
||
> Orchis × bergonii Nanteuil: 9.80%
|
||
> Orchis italica Poir. : 9.66%
|
||
> Orchis × angusticruris Franch.: 2.76%
|
||
> Orchis × bivonae Tod. : 2.54%
|
||
|
||
```
|
||
|
||
|