delcenjo commited on
Commit
bdee883
·
verified ·
1 Parent(s): 530ebeb

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +12 -16
README.md CHANGED
@@ -13,22 +13,18 @@ tags:
13
 
14
  # Flower Image Classifier (ResNet-18, transfer learning)
15
 
16
- 🚀 **Try the live demo:** [flower-classifier-demo Space](https://huggingface.co/spaces/delcenjo/flower-classifier-demo) — upload a photo and get a prediction.
17
- 📦 **Code on GitHub:** [delcenjo/flower-image-classifier](https://github.com/delcenjo/flower-image-classifier)
18
 
19
- A small image classifier that recognises five flower species — **daisy,
20
- dandelion, roses, sunflowers, tulips** using transfer learning on a
21
- pre-trained ResNet-18. The ImageNet backbone is frozen and only a new
22
- classification head is trained, 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 128×128, normalised with ImageNet statistics
26
- - **Test accuracy:** ~0.77 (5 balanced classes; random baseline 0.20)
27
- - **Dataset:** TensorFlow Flowers (~3,670 images)
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 128×128 with a frozen backbone, so accuracy is
67
- modest. Unfreezing the last ResNet block and training at 224×224 would improve it.
 
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.