Hugging Face's logo

dima806
/
house-plant-image-detection

Image Classification
Safetensors
vit
Model card Files Community
dima806 commited on
Commit
a9fe439
·
verified ·
1 Parent(s): 92d2126

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +53 -53
README.md CHANGED
@@ -5,66 +5,66 @@ metrics:
5
  - f1
6
  ---
7
 
8
- Predicts house plant type based on an image.
9
-
10
- ![image/png](https://cdn-uploads.huggingface.co/production/uploads/6449300e3adf50d864095b90/xUlPiU6O5KN6jtW9aBmzd.png)
11
 
12
  See https://www.kaggle.com/code/dima806/house-plant-image-detection-vit for more details.
13
 
 
 
14
  ```
15
  Classification report:
16
 
17
  precision recall f1-score support
18
 
19
- African Violet (Saintpaulia ionantha) 0.0000 0.0000 0.0000 27
20
- Aloe Vera 0.0000 0.0000 0.0000 27
21
- Anthurium (Anthurium andraeanum) 0.0278 0.0370 0.0317 27
22
- Areca Palm (Dypsis lutescens) 0.0000 0.0000 0.0000 27
23
- Asparagus Fern (Asparagus setaceus) 0.0000 0.0000 0.0000 27
24
- Begonia (Begonia spp.) 0.0000 0.0000 0.0000 26
25
- Bird of Paradise (Strelitzia reginae) 0.0000 0.0000 0.0000 27
26
- Birds Nest Fern (Asplenium nidus) 0.0818 0.3333 0.1314 27
27
- Boston Fern (Nephrolepis exaltata) 0.0000 0.0000 0.0000 26
28
- Calathea 0.0000 0.0000 0.0000 26
29
- Cast Iron Plant (Aspidistra elatior) 0.0000 0.0000 0.0000 26
30
- Chinese Money Plant (Pilea peperomioides) 0.1408 0.3846 0.2062 26
31
- Chinese evergreen (Aglaonema) 0.0164 0.0385 0.0230 26
32
- Christmas Cactus (Schlumbergera bridgesii) 0.0000 0.0000 0.0000 27
33
- Chrysanthemum 0.0000 0.0000 0.0000 26
34
- Ctenanthe 0.0000 0.0000 0.0000 26
35
- Daffodils (Narcissus spp.) 0.0000 0.0000 0.0000 27
36
- Dracaena 0.0000 0.0000 0.0000 27
37
- Dumb Cane (Dieffenbachia spp.) 0.0000 0.0000 0.0000 27
38
- Elephant Ear (Alocasia spp.) 0.0000 0.0000 0.0000 26
39
- English Ivy (Hedera helix) 0.0000 0.0000 0.0000 27
40
- Hyacinth (Hyacinthus orientalis) 0.7500 0.4444 0.5581 27
41
- Iron Cross begonia (Begonia masoniana) 0.2812 0.3462 0.3103 26
42
- Jade plant (Crassula ovata) 0.0000 0.0000 0.0000 27
43
- Kalanchoe 0.0789 0.1111 0.0923 27
44
- Lilium (Hemerocallis) 0.1000 0.0385 0.0556 26
45
- Lily of the valley (Convallaria majalis) 0.2368 0.3333 0.2769 27
46
- Money Tree (Pachira aquatica) 0.0000 0.0000 0.0000 27
47
- Monstera Deliciosa (Monstera deliciosa) 0.0000 0.0000 0.0000 26
48
- Orchid 0.0000 0.0000 0.0000 26
49
- Parlor Palm (Chamaedorea elegans) 0.0000 0.0000 0.0000 27
50
- Peace lily 0.0000 0.0000 0.0000 27
51
- Poinsettia (Euphorbia pulcherrima) 0.0476 0.0385 0.0426 26
52
- Polka Dot Plant (Hypoestes phyllostachya) 0.1212 0.1538 0.1356 26
53
- Ponytail Palm (Beaucarnea recurvata) 0.0769 0.1154 0.0923 26
54
- Pothos (Ivy arum) 0.0000 0.0000 0.0000 26
55
- Prayer Plant (Maranta leuconeura) 0.0408 0.0741 0.0526 27
56
- Rattlesnake Plant (Calathea lancifolia) 0.0704 0.1852 0.1020 27
57
- Rubber Plant (Ficus elastica) 0.0000 0.0000 0.0000 26
58
- Sago Palm (Cycas revoluta) 0.0000 0.0000 0.0000 26
59
- Schefflera 0.0000 0.0000 0.0000 27
60
- Snake plant (Sanseviera) 0.2000 0.7037 0.3115 27
61
- Tradescantia 0.1111 0.0385 0.0571 26
62
- Tulip 0.2500 0.1111 0.1538 27
63
- Venus Flytrap 0.0000 0.0000 0.0000 27
64
- Yucca 0.0000 0.0000 0.0000 27
65
- ZZ Plant (Zamioculcas zamiifolia) 0.1780 0.8077 0.2917 26
66
 
67
- accuracy 0.0913 1248
68
- macro avg 0.0598 0.0914 0.0622 1248
69
- weighted avg 0.0599 0.0913 0.0623 1248
70
  ```
 
5
  - f1
6
  ---
7
 
8
+ Predicts house plant type with about 76% accuracy based on an image.
 
 
9
 
10
  See https://www.kaggle.com/code/dima806/house-plant-image-detection-vit for more details.
11
 
12
+ ![image/png](https://cdn-uploads.huggingface.co/production/uploads/6449300e3adf50d864095b90/twVMo96Gdl1u1pzS1I_kP.png)
13
+
14
  ```
15
  Classification report:
16
 
17
  precision recall f1-score support
18
 
19
+ African Violet (Saintpaulia ionantha) 0.9381 0.9292 0.9336 212
20
+ Aloe Vera 0.9072 0.8263 0.8649 213
21
+ Anthurium (Anthurium andraeanum) 0.9383 0.7136 0.8107 213
22
+ Areca Palm (Dypsis lutescens) 0.8167 0.4623 0.5904 212
23
+ Asparagus Fern (Asparagus setaceus) 0.9630 0.7358 0.8342 212
24
+ Begonia (Begonia spp.) 0.8571 0.0283 0.0548 212
25
+ Bird of Paradise (Strelitzia reginae) 0.9869 0.7123 0.8274 212
26
+ Birds Nest Fern (Asplenium nidus) 0.8559 0.9202 0.8869 213
27
+ Boston Fern (Nephrolepis exaltata) 0.8798 0.9670 0.9213 212
28
+ Calathea 0.3815 0.4836 0.4265 213
29
+ Cast Iron Plant (Aspidistra elatior) 0.9329 0.6526 0.7680 213
30
+ Chinese Money Plant (Pilea peperomioides) 0.5072 0.9953 0.6719 213
31
+ Chinese evergreen (Aglaonema) 0.6667 0.2830 0.3974 212
32
+ Christmas Cactus (Schlumbergera bridgesii) 1.0000 0.8538 0.9211 212
33
+ Chrysanthemum 0.9382 0.7840 0.8542 213
34
+ Ctenanthe 0.6093 0.4319 0.5055 213
35
+ Daffodils (Narcissus spp.) 0.9805 0.9481 0.9640 212
36
+ Dracaena 0.6408 0.7371 0.6856 213
37
+ Dumb Cane (Dieffenbachia spp.) 0.4463 0.8592 0.5875 213
38
+ Elephant Ear (Alocasia spp.) 0.9813 0.4953 0.6583 212
39
+ English Ivy (Hedera helix) 0.7589 0.9057 0.8258 212
40
+ Hyacinth (Hyacinthus orientalis) 0.9193 0.9670 0.9425 212
41
+ Iron Cross begonia (Begonia masoniana) 0.6852 0.9812 0.8069 213
42
+ Jade plant (Crassula ovata) 0.8052 0.5849 0.6776 212
43
+ Kalanchoe 0.6090 0.7606 0.6764 213
44
+ Lilium (Hemerocallis) 0.8722 0.9340 0.9021 212
45
+ Lily of the valley (Convallaria majalis) 0.9327 0.9811 0.9563 212
46
+ Money Tree (Pachira aquatica) 0.6667 0.0376 0.0711 213
47
+ Monstera Deliciosa (Monstera deliciosa) 0.9623 0.7217 0.8248 212
48
+ Orchid 1.0000 0.8685 0.9296 213
49
+ Parlor Palm (Chamaedorea elegans) 0.5155 0.9434 0.6667 212
50
+ Peace lily 0.9926 0.6291 0.7701 213
51
+ Poinsettia (Euphorbia pulcherrima) 0.9048 0.9858 0.9436 212
52
+ Polka Dot Plant (Hypoestes phyllostachya) 0.6581 0.9671 0.7833 213
53
+ Ponytail Palm (Beaucarnea recurvata) 0.8626 0.7406 0.7970 212
54
+ Pothos (Ivy arum) 0.8727 0.2264 0.3596 212
55
+ Prayer Plant (Maranta leuconeura) 0.5312 0.7217 0.6120 212
56
+ Rattlesnake Plant (Calathea lancifolia) 0.6596 0.8868 0.7565 212
57
+ Rubber Plant (Ficus elastica) 0.8624 0.8826 0.8724 213
58
+ Sago Palm (Cycas revoluta) 1.0000 0.9481 0.9734 212
59
+ Schefflera 0.4687 0.8821 0.6121 212
60
+ Snake plant (Sanseviera) 0.7833 0.8826 0.8300 213
61
+ Tradescantia 0.9048 0.8066 0.8529 212
62
+ Tulip 0.9476 0.9387 0.9431 212
63
+ Venus Flytrap 1.0000 0.8915 0.9426 212
64
+ Yucca 0.8318 0.8357 0.8337 213
65
+ ZZ Plant (Zamioculcas zamiifolia) 0.5562 0.9575 0.7036 212
66
 
67
+ accuracy 0.7593 9983
68
+ macro avg 0.8041 0.7593 0.7453 9983
69
+ weighted avg 0.8040 0.7593 0.7453 9983
70
  ```