Skip to content

Files

Failed to load latest commit information.

Latest commit

 Cannot retrieve latest commit at this time.

History

History
 
 

repvgg

Folders and files

NameName
Last commit message
Last commit date

parent directory

..
 
 
 
 

candle-repvgg

RepVGG: Making VGG-style ConvNets Great Again.

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

loaded image Tensor[dims 3, 224, 224; f32]
model built
mountain bike, all-terrain bike, off-roader: 61.70%
bicycle-built-for-two, tandem bicycle, tandem: 33.14%
unicycle, monocycle     : 4.88%
crash helmet            : 0.15%
moped                   : 0.04%