huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-16 10:38:54 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
410c89f72a0ab22a299d02d24f505a50522faaa2
candle/candle-examples/examples/mobileone/README.md
Jani Monoses 5270224f40 Add MobileOne model. (#1595) 
* Add MobileOne model.

* Clippy fixes

* Remove a comment.

---------

Co-authored-by: laurent <laurent.mazare@gmail.com>
2024-01-16 06:34:16 +01:00

714 B
Raw Blame History

candle-mobileone

MobileOne: An Improved One millisecond Mobile Backbone.

This candle implementation uses a pre-trained MobileOne 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 mobileone --release -- --image candle-examples/examples/yolo-v8/assets/bike.jpg --which s2

loaded image Tensor[dims 3, 224, 224; f32]
model built
mountain bike, all-terrain bike, off-roader: 79.33%
bicycle-built-for-two, tandem bicycle, tandem: 15.32%
crash helmet            : 2.58%
unicycle, monocycle     : 1.70%
alp                     : 0.21%