Flower Classifier

finetuned from PyTorch ResNet18

Flower Label

0: Lotus
1: Orchid
2: Tulip
3: Sunflower
4: Lilly

How to Use:


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()