Hugging Face's logo

dima806
/
house-plant-image-detection

Image Classification
Safetensors
vit
Model card Files Community
dima806 commited on
Commit
31c6487
•
1 Parent(s): 6c7c10c

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +52 -52
README.md CHANGED
@@ -5,66 +5,66 @@ metrics:
5
  - f1
6
  ---
7
 
8
- Predicts house plant type with about 84% 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/1N1R_HhOTTsWLaDOK-CZX.png)
13
 
14
  ```
15
  Classification report:
16
 
17
  precision recall f1-score support
18
 
19
- African Violet (Saintpaulia ionantha) 0.9623 0.9444 0.9533 216
20
- Aloe Vera 0.9367 0.9628 0.9495 215
21
- Anthurium (Anthurium andraeanum) 0.9337 0.7824 0.8514 216
22
- Areca Palm (Dypsis lutescens) 0.9574 0.6250 0.7563 216
23
- Asparagus Fern (Asparagus setaceus) 0.9950 0.9167 0.9542 216
24
- Begonia (Begonia spp.) 1.0000 0.1204 0.2149 216
25
- Bird of Paradise (Strelitzia reginae) 0.9767 0.7778 0.8660 216
26
- Birds Nest Fern (Asplenium nidus) 0.9406 0.9581 0.9493 215
27
- Boston Fern (Nephrolepis exaltata) 0.9188 0.9954 0.9556 216
28
- Calathea 0.5813 0.4326 0.4960 215
29
- Cast Iron Plant (Aspidistra elatior) 0.8915 0.8750 0.8832 216
30
- Chinese Money Plant (Pilea peperomioides) 0.7709 0.9815 0.8635 216
31
- Chinese evergreen (Aglaonema) 0.9157 0.3519 0.5084 216
32
- Christmas Cactus (Schlumbergera bridgesii) 0.9951 0.9442 0.9690 215
33
- Chrysanthemum 0.9802 0.9209 0.9496 215
34
- Ctenanthe 0.7581 0.6528 0.7015 216
35
- Daffodils (Narcissus spp.) 0.9680 0.9860 0.9770 215
36
- Dracaena 0.7452 0.7176 0.7311 216
37
- Dumb Cane (Dieffenbachia spp.) 0.5726 0.9535 0.7155 215
38
- Elephant Ear (Alocasia spp.) 0.8919 0.7639 0.8229 216
39
- English Ivy (Hedera helix) 0.8171 0.9722 0.8879 216
40
- Hyacinth (Hyacinthus orientalis) 0.9509 0.9861 0.9682 216
41
- Iron Cross begonia (Begonia masoniana) 0.6817 0.9860 0.8061 215
42
- Jade plant (Crassula ovata) 0.7524 0.7176 0.7346 216
43
- Kalanchoe 0.7293 0.7767 0.7523 215
44
- Lilium (Hemerocallis) 0.9152 0.9535 0.9339 215
45
- Lily of the valley (Convallaria majalis) 0.9591 0.9769 0.9679 216
46
- Money Tree (Pachira aquatica) 0.8889 0.2222 0.3556 216
47
- Monstera Deliciosa (Monstera deliciosa) 0.9420 0.9028 0.9220 216
48
- Orchid 0.9766 0.9721 0.9744 215
49
- Parlor Palm (Chamaedorea elegans) 0.6646 0.9767 0.7910 215
50
- Peace lily 0.9570 0.8241 0.8856 216
51
- Poinsettia (Euphorbia pulcherrima) 0.9430 1.0000 0.9707 215
52
- Polka Dot Plant (Hypoestes phyllostachya) 0.7918 0.9861 0.8784 216
53
- Ponytail Palm (Beaucarnea recurvata) 0.9583 0.8519 0.9020 216
54
- Pothos (Ivy arum) 0.9517 0.6389 0.7645 216
55
- Prayer Plant (Maranta leuconeura) 0.6729 0.8419 0.7479 215
56
- Rattlesnake Plant (Calathea lancifolia) 0.7794 0.9860 0.8706 215
57
- Rubber Plant (Ficus elastica) 0.8729 0.9537 0.9115 216
58
- Sago Palm (Cycas revoluta) 0.9772 0.9953 0.9862 215
59
- Schefflera 0.5149 0.9583 0.6699 216
60
- Snake plant (Sanseviera) 0.8829 0.9116 0.8970 215
61
- Tradescantia 0.9118 0.8611 0.8857 216
62
- Tulip 0.9767 0.9767 0.9767 215
63
- Venus Flytrap 1.0000 0.9167 0.9565 216
64
- Yucca 0.8038 0.9861 0.8857 216
65
- ZZ Plant (Zamioculcas zamiifolia) 0.7747 0.9074 0.8358 216
66
 
67
- accuracy 0.8446 10134
68
- macro avg 0.8668 0.8447 0.8338 10134
69
- weighted avg 0.8668 0.8446 0.8337 10134
70
  ```
 
5
  - f1
6
  ---
7
 
8
+ Predicts house plant type with about 88% 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/6G0biOkhS2lgAW0WveeVT.png)
13
 
14
  ```
15
  Classification report:
16
 
17
  precision recall f1-score support
18
 
19
+ African Violet (Saintpaulia ionantha) 0.9813 0.9767 0.9790 215
20
+ Aloe Vera 0.9638 0.9861 0.9748 216
21
+ Anthurium (Anthurium andraeanum) 0.9448 0.7917 0.8615 216
22
+ Areca Palm (Dypsis lutescens) 0.9482 0.8472 0.8949 216
23
+ Asparagus Fern (Asparagus setaceus) 0.9854 0.9352 0.9596 216
24
+ Begonia (Begonia spp.) 0.9744 0.1759 0.2980 216
25
+ Bird of Paradise (Strelitzia reginae) 1.0000 0.8194 0.9008 216
26
+ Birds Nest Fern (Asplenium nidus) 0.9498 0.9630 0.9563 216
27
+ Boston Fern (Nephrolepis exaltata) 0.9464 0.9860 0.9658 215
28
+ Calathea 0.8102 0.5139 0.6289 216
29
+ Cast Iron Plant (Aspidistra elatior) 0.8071 0.9491 0.8723 216
30
+ Chinese Money Plant (Pilea peperomioides) 0.8898 0.9767 0.9313 215
31
+ Chinese evergreen (Aglaonema) 0.9688 0.4306 0.5962 216
32
+ Christmas Cactus (Schlumbergera bridgesii) 1.0000 0.9398 0.9690 216
33
+ Chrysanthemum 0.9638 0.9861 0.9748 216
34
+ Ctenanthe 0.7532 0.8194 0.7849 216
35
+ Daffodils (Narcissus spp.) 0.9862 0.9954 0.9908 216
36
+ Dracaena 0.8471 0.6698 0.7481 215
37
+ Dumb Cane (Dieffenbachia spp.) 0.6426 0.9535 0.7678 215
38
+ Elephant Ear (Alocasia spp.) 0.8026 0.8698 0.8348 215
39
+ English Ivy (Hedera helix) 0.9025 0.9907 0.9446 215
40
+ Hyacinth (Hyacinthus orientalis) 0.9907 0.9861 0.9884 216
41
+ Iron Cross begonia (Begonia masoniana) 0.6941 0.9769 0.8115 216
42
+ Jade plant (Crassula ovata) 0.7143 0.8796 0.7884 216
43
+ Kalanchoe 0.7321 0.7083 0.7200 216
44
+ Lilium (Hemerocallis) 0.9550 0.9860 0.9703 215
45
+ Lily of the valley (Convallaria majalis) 0.9679 0.9769 0.9724 216
46
+ Money Tree (Pachira aquatica) 0.9292 0.4861 0.6383 216
47
+ Monstera Deliciosa (Monstera deliciosa) 0.9206 0.9163 0.9184 215
48
+ Orchid 0.9953 0.9953 0.9953 215
49
+ Parlor Palm (Chamaedorea elegans) 0.8160 0.9488 0.8774 215
50
+ Peace lily 0.9423 0.9116 0.9267 215
51
+ Poinsettia (Euphorbia pulcherrima) 0.9640 0.9907 0.9772 216
52
+ Polka Dot Plant (Hypoestes phyllostachya) 0.8583 0.9815 0.9158 216
53
+ Ponytail Palm (Beaucarnea recurvata) 0.9806 0.9395 0.9596 215
54
+ Pothos (Ivy arum) 0.9557 0.8981 0.9260 216
55
+ Prayer Plant (Maranta leuconeura) 0.8008 0.9120 0.8528 216
56
+ Rattlesnake Plant (Calathea lancifolia) 0.8669 0.9954 0.9267 216
57
+ Rubber Plant (Ficus elastica) 0.8932 0.9721 0.9310 215
58
+ Sago Palm (Cycas revoluta) 0.9954 1.0000 0.9977 216
59
+ Schefflera 0.6753 0.9630 0.7939 216
60
+ Snake plant (Sanseviera) 0.9714 0.9488 0.9600 215
61
+ Tradescantia 0.9561 0.9116 0.9333 215
62
+ Tulip 0.9954 1.0000 0.9977 215
63
+ Venus Flytrap 0.9950 0.9349 0.9640 215
64
+ Yucca 0.7626 0.9815 0.8583 216
65
+ ZZ Plant (Zamioculcas zamiifolia) 0.8590 0.9028 0.8804 216
66
 
67
+ accuracy 0.8867 10134
68
+ macro avg 0.8990 0.8868 0.8791 10134
69
+ weighted avg 0.8990 0.8867 0.8790 10134
70
  ```