Downloads · 30 days
0
resnet18-flower-classification
Dimasnoufal/resnet18-flower-classification
resnet18-flower-classification is a machine learning model from Dimasnoufal. Use it for the machine learning task on the model card, and read the license before you ship it in a product.
<h1Flower Classifier</h1 finetuned from PyTorch ResNet18
Access
Public
Updated Feb 20, 2024
Repo size
46.8 MB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.pth46.8 MB · 100%
At a glance
- Access
- Public
- Created
- Feb 20, 2024
- Updated
- Feb 20, 2024
- SHA
- 754fe13a
Model card
From the Hugging Face model README
<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>
- Created
- Feb 20, 2024
- Updated
- Feb 20, 2024