To use ResNet in your application, take a look at the official Burn implementation available on GitHub! It closely follows this tutorial's implementation but further extends it to provide an easy interface to load the pre-trained weights for the whole ResNet family of models.
- Download the ResNet-18 pre-trained weights from
torchvision
wget https://download.pytorch.org/models/resnet18-f37072fd.pth
- Download a sample image for inference
wget https://upload.wikimedia.org/wikipedia/commons/2/26/YellowLabradorLooking_new.jpg
- Run the example
cargo run --release YellowLabradorLooking_new.jpg