mirror of
https://github.com/huggingface/candle.git
synced 2025-06-17 02:58:50 +00:00
Add a readme for the resnet example. (#1129)
This commit is contained in:
@ -166,6 +166,7 @@ If you have an addition to this list, please submit a pull request.
|
||||
- DINOv2.
|
||||
- ConvMixer.
|
||||
- EfficientNet.
|
||||
- ResNet-18/34.
|
||||
- yolo-v3.
|
||||
- yolo-v8.
|
||||
- Segment-Anything Model (SAM).
|
||||
|
19
candle-examples/examples/resnet/README.md
Normal file
19
candle-examples/examples/resnet/README.md
Normal file
@ -0,0 +1,19 @@
|
||||
# candle-resnet
|
||||
|
||||
A candle implementation of inference using a pre-trained [ResNet](https://arxiv.org/abs/1512.03385).
|
||||
This uses a classification head trained on the ImageNet dataset and returns the
|
||||
probabilities for the top-5 classes.
|
||||
|
||||
## Running an example
|
||||
|
||||
```
|
||||
$ cargo run --example resnet --release -- --image tiger.jpg
|
||||
|
||||
loaded image Tensor[dims 3, 224, 224; f32]
|
||||
model built
|
||||
tiger, Panthera tigris : 90.21%
|
||||
tiger cat : 8.93%
|
||||
lion, king of beasts, Panthera leo: 0.35%
|
||||
leopard, Panthera pardus: 0.16%
|
||||
jaguar, panther, Panthera onca, Felis onca: 0.09%
|
||||
```
|
Reference in New Issue
Block a user