mirror of
https://github.com/huggingface/candle.git
synced 2025-06-16 10:38:54 +00:00
dece37c6f4
* 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>
14 lines
674 B
Markdown
14 lines
674 B
Markdown
## 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:
|
|
|
|
```bash
|
|
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).
|