laggui / resnet-burn

ResNet implementation with Burn

👉 Check out the full tutorial 👈

ResNet + Burn ✍️

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.

Example Usage

1. Download the ResNet-18 pre-trained weights from torchvision

wget https://download.pytorch.org/models/resnet18-f37072fd.pth

2. Download a sample image for inference

wget https://upload.wikimedia.org/wikipedia/commons/2/26/YellowLabradorLooking_new.jpg

3. Run the example

cargo run --release YellowLabradorLooking_new.jpg