diff options
author | Laurent Mazare <laurent.mazare@gmail.com> | 2023-10-19 09:58:50 +0100 |
---|---|---|
committer | GitHub <noreply@github.com> | 2023-10-19 09:58:50 +0100 |
commit | 6f76383f3885c9ba1e5aa39a7ee0bef994069e1c (patch) | |
tree | 32fb0d36ac69dd330c975989ad316f445e638bae | |
parent | 8e773cc0c61af4ee9015aedc1e71398185c84f25 (diff) | |
download | candle-6f76383f3885c9ba1e5aa39a7ee0bef994069e1c.tar.gz candle-6f76383f3885c9ba1e5aa39a7ee0bef994069e1c.tar.bz2 candle-6f76383f3885c9ba1e5aa39a7ee0bef994069e1c.zip |
Add a readme for the resnet example. (#1129)
-rw-r--r-- | README.md | 1 | ||||
-rw-r--r-- | candle-examples/examples/resnet/README.md | 19 |
2 files changed, 20 insertions, 0 deletions
@@ -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). diff --git a/candle-examples/examples/resnet/README.md b/candle-examples/examples/resnet/README.md new file mode 100644 index 00000000..df934773 --- /dev/null +++ b/candle-examples/examples/resnet/README.md @@ -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% +``` |