Dimasnoufal
/

resnet18-flower-classification

Model card Files Files and versions Community

YAML Metadata Warning: empty or missing yaml metadata in repo card (https://huggingface.co/docs/hub/model-cards#model-card-metadata)

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

Downloads last month: 0

Unable to determine this model's library. Check the docs .