Skip to content

Latest commit

 

History

History
28 lines (19 loc) · 851 Bytes

README.md

File metadata and controls

28 lines (19 loc) · 851 Bytes
Raw

👉 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
  1. Download a sample image for inference
wget https://upload.wikimedia.org/wikipedia/commons/2/26/YellowLabradorLooking_new.jpg
  1. Run the example
cargo run --release YellowLabradorLooking_new.jpg