Upload README.md with huggingface_hub
Browse files
README.md
CHANGED
|
@@ -13,22 +13,18 @@ tags:
|
|
| 13 |
|
| 14 |
# Flower Image Classifier (ResNet-18, transfer learning)
|
| 15 |
|
| 16 |
-
|
| 17 |
-
|
| 18 |
|
| 19 |
-
A small image classifier that recognises five flower species
|
| 20 |
-
|
| 21 |
-
|
| 22 |
-
|
| 23 |
|
| 24 |
-
-
|
| 25 |
-
-
|
| 26 |
-
-
|
| 27 |
-
-
|
| 28 |
-
|
| 29 |
-
## Files
|
| 30 |
-
|
| 31 |
-
`flower_classifier.pt` is a checkpoint: `{"model_state": state_dict, "classes": [...]}`.
|
| 32 |
|
| 33 |
## Usage
|
| 34 |
|
|
@@ -63,5 +59,5 @@ print(classes[int(probs.argmax())], float(probs.max()))
|
|
| 63 |
|
| 64 |
## Limitations
|
| 65 |
|
| 66 |
-
Trained on a small dataset at
|
| 67 |
-
modest. Unfreezing the last ResNet block and training at
|
|
|
|
| 13 |
|
| 14 |
# Flower Image Classifier (ResNet-18, transfer learning)
|
| 15 |
|
| 16 |
+
Try the live demo: https://huggingface.co/spaces/delcenjo/flower-classifier-demo
|
| 17 |
+
Code on GitHub: https://github.com/delcenjo/flower-image-classifier
|
| 18 |
|
| 19 |
+
A small image classifier that recognises five flower species (daisy, dandelion,
|
| 20 |
+
roses, sunflowers, tulips) using transfer learning on a pre-trained ResNet-18.
|
| 21 |
+
The ImageNet backbone is frozen and only a new classification head is trained,
|
| 22 |
+
so it runs well on CPU.
|
| 23 |
|
| 24 |
+
- Architecture: ResNet-18 (ImageNet weights), final layer replaced with a 5-class head
|
| 25 |
+
- Input: RGB image resized to 128x128, normalised with ImageNet statistics
|
| 26 |
+
- Test accuracy: about 0.77 (5 balanced classes; random baseline 0.20)
|
| 27 |
+
- Dataset: TensorFlow Flowers (about 3,670 images)
|
|
|
|
|
|
|
|
|
|
|
|
|
| 28 |
|
| 29 |
## Usage
|
| 30 |
|
|
|
|
| 59 |
|
| 60 |
## Limitations
|
| 61 |
|
| 62 |
+
Trained on a small dataset at 128x128 with a frozen backbone, so accuracy is
|
| 63 |
+
modest. Unfreezing the last ResNet block and training at 224x224 would improve it.
|