| <h1>Flower Classifier</h1> | |
| finetuned from PyTorch ResNet18 | |
| <h2>Flower Label</h2> | |
| 0: Lotus<br> | |
| 1: Orchid<br> | |
| 2: Tulip<br> | |
| 3: Sunflower<br> | |
| 4: Lilly<br> | |
| <h2>How to Use:</h2> | |
| <pre><code class="python"> | |
| import torchvision | |
| from torchvision.models import resnet18 | |
| model = resnet18(weights=None) | |
| model_weights_path = 'resnet18_flower_classification.pth' | |
| model.load_state_dict(torch.load(model_weights_path)) | |
| model.eval() | |
| </pre> |