Skip to content

laggui/resnet-burn

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

9 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

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

About

ResNet implementation with Burn

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published