huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-22 12:28:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add FastViT model. (#2444)

Browse Source
This commit is contained in:
Jani Monoses
2024-08-23 17:06:54 +03:00
committed by GitHub
parent 2ec8729d51
commit ccdbe87639
4 changed files with 635 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
candle-examples/examples/fastvit/README.md Normal file
View File

@ -0,0 +1,20 @@
# candle-fastvit
[FastViT: A Fast Hybrid Vision Transformer using Structural Reparameterization](https://arxiv.org/abs/2303.14189).
This candle implementation uses a pre-trained FastViT network for inference. The
classification head has been trained on the ImageNet dataset and returns the
probabilities for the top-5 classes.
## Running an example
```
$ cargo run --example fastvit --release -- --image candle-examples/examples/yolo-v8/assets/bike.jpg --which sa12
loaded image Tensor[dims 3, 256, 256; f32]
model built
mountain bike, all-terrain bike, off-roader: 43.45%
bicycle-built-for-two, tandem bicycle, tandem: 14.16%
unicycle, monocycle : 4.12%
crash helmet : 2.26%
alp : 1.40%
```