huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-15 10:26:33 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
dece37c6f4d9c5a52caf59a003afa6ba33034fe3
candle/candle-examples/examples/vgg/README.md
drbh dece37c6f4 feat: implement VGG13, VGG16 and VGG19 (#1211) 
* feat: implement VGG13, VGG16 and VGG19

* Cosmetic fixes.

* More cosmetic tweaks + avoid re-loading the weights on each final layer.

---------

Co-authored-by: Laurent <laurent.mazare@gmail.com>
2023-10-29 06:10:23 +00:00

674 B
Raw Blame History

VGG Model Implementation

This example demonstrates the implementation of VGG models (VGG13, VGG16, VGG19) using the Candle library.

The VGG models are defined in candle-transformers/src/models/vgg.rs. The main function in candle-examples/examples/vgg/main.rs loads an image, selects the VGG model based on the provided argument, and applies the model to the loaded image.

You can run the example with the following command:

cargo run --example vgg --release -- --image ../yolo-v8/assets/bike.jpg --which vgg13

In the command above, --image specifies the path to the image file and --which specifies the VGG model to use (vgg13, vgg16, or vgg19).