https://github.com/laggui/resnet-burn
ResNet implementation with Burn
https://github.com/laggui/resnet-burn
Last synced: 17 days ago
JSON representation
ResNet implementation with Burn
- Host: GitHub
- URL: https://github.com/laggui/resnet-burn
- Owner: laggui
- License: mit
- Created: 2024-02-14T15:52:48.000Z (over 1 year ago)
- Default Branch: main
- Last Pushed: 2024-02-14T20:44:54.000Z (over 1 year ago)
- Last Synced: 2024-02-15T17:38:07.987Z (over 1 year ago)
- Language: Rust
- Size: 16.6 KB
- Stars: 9
- Watchers: 1
- Forks: 0
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
- License: LICENSE
Awesome Lists containing this project
README
# **👉 [Check out the full tutorial](https://dev.to/laggui/transitioning-from-pytorch-to-burn-45m) 👈**
## ResNet + Burn ✍️
To use ResNet in your application, take a look at the official Burn implementation
[available on GitHub](https://github.com/tracel-ai/models/tree/main/resnet-burn)! 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`
```sh
wget https://download.pytorch.org/models/resnet18-f37072fd.pth
```
2. Download a sample image for inference
```sh
wget https://upload.wikimedia.org/wikipedia/commons/2/26/YellowLabradorLooking_new.jpg
```
3. Run the example
```sh
cargo run --release YellowLabradorLooking_new.jpg
```