Hugging Face's logo

dima806
/
mushrooms_image_detection

Image Classification
Transformers
PyTorch
Safetensors
vit
Model card Files Community
dima806 commited on
Commit
d563703
·
verified ·
1 Parent(s): 2a7e48a

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +105 -103
README.md CHANGED
@@ -8,113 +8,115 @@ Returns mushroom type given an image.
8
 
9
  See https://www.kaggle.com/code/dima806/mushrooms-image-detection-vit for more details.
10
 
 
 
11
  ```
12
  Classification report:
13
 
14
  precision recall f1-score support
15
 
16
- Urnula craterium 0.9861 0.9709 0.9784 584
17
- Leccinum albostipitatum 0.7789 0.8870 0.8295 584
18
- Lactarius deliciosus 0.9245 0.7551 0.8313 584
19
- Clitocybe nebularis 0.9477 0.9315 0.9396 584
20
- Hypholoma fasciculare 0.8954 0.7329 0.8060 584
21
- Lactarius torminosus 0.8300 0.9195 0.8725 584
22
- Lycoperdon perlatum 0.9418 0.9692 0.9553 584
23
- Verpa bohemica 0.9948 0.9914 0.9931 584
24
- Schizophyllum commune 0.8908 0.9794 0.9330 583
25
- Leccinum aurantiacum 0.6299 0.5274 0.5741 584
26
- Phellinus igniarius 0.8514 0.8339 0.8426 584
27
- Suillus luteus 0.7542 0.3105 0.4399 583
28
- Coltricia perennis 0.9712 0.9247 0.9474 584
29
- Cetraria islandica 0.9863 0.9863 0.9863 584
30
- Amanita muscaria 0.9678 0.9777 0.9727 584
31
- Pholiota aurivella 0.9229 0.9435 0.9331 584
32
- Trichaptum biforme 0.8301 0.8714 0.8502 583
33
- Artomyces pyxidatus 1.0000 0.9983 0.9991 584
34
- Calocera viscosa 1.0000 0.9983 0.9991 584
35
- Sarcosoma globosum 0.9592 0.9658 0.9625 584
36
- Evernia prunastri 0.8209 0.8870 0.8527 584
37
- Laetiporus sulphureus 0.9595 0.9726 0.9660 584
38
- Lobaria pulmonaria 0.9563 0.9743 0.9652 584
39
- Bjerkandera adusta 0.8187 0.8130 0.8158 583
40
- Vulpicida pinastri 0.9666 0.9914 0.9788 583
41
- Imleria badia 0.7430 0.7226 0.7326 584
42
- Evernia mesomorpha 0.9072 0.9057 0.9064 583
43
- Physcia adscendens 0.7900 0.7860 0.7880 584
44
- Coprinellus micaceus 0.8919 0.8919 0.8919 583
45
- Armillaria borealis 0.9414 0.4675 0.6247 584
46
- Trametes ochracea 0.7947 0.5634 0.6593 584
47
- Cantharellus cibarius 0.8896 0.9658 0.9261 584
48
- Pseudevernia furfuracea 0.8879 0.8271 0.8564 584
49
- Tremella mesenterica 0.9845 0.9795 0.9820 584
50
- Gyromitra infula 0.9620 0.9538 0.9579 584
51
- Leccinum versipelle 0.7403 0.6884 0.7134 584
52
- Mutinus ravenelii 0.9983 1.0000 0.9991 584
53
- Pholiota squarrosa 0.7775 0.9332 0.8482 584
54
- Amanita rubescens 0.8494 0.9091 0.8782 583
55
- Amanita pantherina 0.9407 0.8699 0.9039 584
56
- Sarcoscypha austriaca 0.9965 0.9897 0.9931 583
57
- Boletus edulis 0.5277 0.9485 0.6781 583
58
- Coprinus comatus 0.9338 0.9897 0.9609 584
59
- Merulius tremellosus 0.8290 0.9400 0.8810 583
60
- Stropharia aeruginosa 0.9731 0.9914 0.9822 583
61
- Cladonia fimbriata 0.9579 0.9743 0.9660 584
62
- Suillus grevillei 0.9207 0.2586 0.4037 584
63
- Apioperdon pyriforme 0.9048 0.9435 0.9237 584
64
- Cerioporus squamosus 0.9387 0.9709 0.9545 584
65
- Leccinum scabrum 0.6923 0.9247 0.7918 584
66
- Rhytisma acerinum 0.9932 0.9966 0.9949 584
67
- Hypholoma lateritium 0.7899 0.9075 0.8446 584
68
- Flammulina velutipes 0.8756 0.8921 0.8838 584
69
- Tricholomopsis rutilans 0.9185 0.7736 0.8399 583
70
- Coprinopsis atramentaria 0.9217 0.9075 0.9146 584
71
- Trametes versicolor 0.8134 0.9194 0.8631 583
72
- Graphis scripta 0.9779 0.9829 0.9804 584
73
- Ganoderma applanatum 0.9294 0.9469 0.9381 584
74
- Phellinus tremulae 0.9223 0.8353 0.8767 583
75
- Peltigera aphthosa 0.9746 0.9846 0.9796 584
76
- Parmelia sulcata 0.8856 0.9297 0.9071 583
77
- Fomitopsis betulina 0.8107 0.9606 0.8793 584
78
- Pleurotus pulmonarius 0.8918 0.8902 0.8910 583
79
- Fomitopsis pinicola 0.9225 0.9777 0.9493 584
80
- Daedaleopsis confragosa 0.7115 0.8784 0.7862 584
81
- Hericium coralloides 0.9965 0.9897 0.9931 583
82
- Trametes hirsuta 0.8563 0.7667 0.8090 583
83
- Coprinellus disseminatus 0.9406 0.9503 0.9454 583
84
- Kuehneromyces mutabilis 0.7233 0.9178 0.8091 584
85
- Pleurotus ostreatus 0.6641 0.8870 0.7595 584
86
- Phlebia radiata 0.9527 0.9315 0.9420 584
87
- Boletus reticulatus 1.0000 0.0223 0.0436 584
88
- Phallus impudicus 0.9964 0.9555 0.9755 584
89
- Macrolepiota procera 0.9714 0.9897 0.9805 584
90
- Fomes fomentarius 0.9025 0.9365 0.9192 583
91
- Suillus granulatus 0.4131 0.9315 0.5723 584
92
- Gyromitra esculenta 0.9321 0.9401 0.9361 584
93
- Xanthoria parietina 0.9603 0.9554 0.9579 583
94
- Nectria cinnabarina 0.9792 0.9675 0.9733 584
95
- Sarcomyxa serotina 0.9649 0.2825 0.4371 584
96
- Inonotus obliquus 0.9522 0.9914 0.9714 583
97
- Panellus stipticus 0.8772 0.7825 0.8271 584
98
- Hypogymnia physodes 0.8489 0.9348 0.8898 583
99
- Hygrophoropsis aurantiaca 0.9174 0.8938 0.9055 584
100
- Cladonia rangiferina 0.9502 0.8818 0.9147 584
101
- Platismatia glauca 0.9413 0.9332 0.9372 584
102
- Calycina citrina 0.9525 0.9966 0.9741 584
103
- Cladonia stellaris 0.9173 0.9692 0.9425 584
104
- Amanita citrina 0.9215 0.9863 0.9528 583
105
- Lepista nuda 0.9628 0.9760 0.9694 584
106
- Gyromitra gigas 0.9672 0.9589 0.9630 584
107
- Crucibulum laeve 0.9114 0.9708 0.9402 583
108
- Daedaleopsis tricolor 0.9275 0.7449 0.8262 584
109
- Stereum hirsutum 0.8534 0.8491 0.8512 583
110
- Paxillus involutus 0.7054 0.9160 0.7970 583
111
- Lactarius turpis 0.9161 0.8596 0.8869 584
112
- Chlorociboria aeruginascens 1.0000 0.9846 0.9922 583
113
- Chondrostereum purpureum 0.9451 0.8836 0.9133 584
114
- Phaeophyscia orbicularis 0.9157 0.7822 0.8437 583
115
- Peltigera praetextata 0.9790 0.9589 0.9689 584
116
 
117
- accuracy 0.8797 58370
118
- macro avg 0.8931 0.8797 0.8728 58370
119
- weighted avg 0.8931 0.8797 0.8728 58370
120
  ```
 
8
 
9
  See https://www.kaggle.com/code/dima806/mushrooms-image-detection-vit for more details.
10
 
11
+ ![image/png](https://cdn-uploads.huggingface.co/production/uploads/6449300e3adf50d864095b90/veq8KASnvJeXh2hCVpcb1.png)
12
+
13
  ```
14
  Classification report:
15
 
16
  precision recall f1-score support
17
 
18
+ Urnula craterium 0.9737 0.9837 0.9787 2335
19
+ Leccinum albostipitatum 0.7604 0.9083 0.8278 2334
20
+ Lactarius deliciosus 0.9254 0.7807 0.8469 2335
21
+ Clitocybe nebularis 0.9330 0.9370 0.9350 2335
22
+ Hypholoma fasciculare 0.9072 0.7537 0.8234 2335
23
+ Lactarius torminosus 0.8531 0.9276 0.8888 2335
24
+ Lycoperdon perlatum 0.9496 0.9675 0.9584 2335
25
+ Verpa bohemica 0.9936 0.9927 0.9931 2335
26
+ Schizophyllum commune 0.8951 0.9756 0.9336 2335
27
+ Leccinum aurantiacum 0.6987 0.5017 0.5840 2334
28
+ Phellinus igniarius 0.8489 0.8180 0.8332 2335
29
+ Suillus luteus 0.7214 0.2861 0.4097 2335
30
+ Coltricia perennis 0.9720 0.9366 0.9540 2335
31
+ Cetraria islandica 0.9793 0.9914 0.9853 2334
32
+ Amanita muscaria 0.9862 0.9790 0.9826 2334
33
+ Pholiota aurivella 0.9190 0.9567 0.9375 2335
34
+ Trichaptum biforme 0.8635 0.8749 0.8692 2335
35
+ Artomyces pyxidatus 0.9996 0.9949 0.9972 2334
36
+ Calocera viscosa 0.9996 0.9991 0.9994 2335
37
+ Sarcosoma globosum 0.9733 0.9675 0.9704 2335
38
+ Evernia prunastri 0.8169 0.8959 0.8546 2335
39
+ Laetiporus sulphureus 0.9581 0.9794 0.9687 2335
40
+ Lobaria pulmonaria 0.9627 0.9824 0.9724 2335
41
+ Bjerkandera adusta 0.8201 0.7867 0.8031 2335
42
+ Vulpicida pinastri 0.9745 0.9824 0.9785 2335
43
+ Imleria badia 0.7476 0.7678 0.7576 2334
44
+ Evernia mesomorpha 0.9155 0.9006 0.9080 2335
45
+ Physcia adscendens 0.8105 0.7935 0.8019 2334
46
+ Coprinellus micaceus 0.8902 0.8887 0.8894 2335
47
+ Armillaria borealis 0.9347 0.5152 0.6643 2335
48
+ Trametes ochracea 0.8031 0.5957 0.6840 2335
49
+ Cantharellus cibarius 0.9021 0.9717 0.9356 2334
50
+ Pseudevernia furfuracea 0.9043 0.8257 0.8632 2335
51
+ Tremella mesenterica 0.9910 0.9906 0.9908 2335
52
+ Gyromitra infula 0.9714 0.9452 0.9581 2335
53
+ Leccinum versipelle 0.7408 0.7379 0.7393 2335
54
+ Mutinus ravenelii 0.9970 1.0000 0.9985 2335
55
+ Pholiota squarrosa 0.7885 0.9195 0.8490 2335
56
+ Amanita rubescens 0.8580 0.9057 0.8812 2334
57
+ Amanita pantherina 0.9411 0.8625 0.9001 2335
58
+ Sarcoscypha austriaca 0.9931 0.9927 0.9929 2334
59
+ Boletus edulis 0.5447 0.9400 0.6897 2335
60
+ Coprinus comatus 0.9476 0.9914 0.9690 2334
61
+ Merulius tremellosus 0.8448 0.9233 0.8823 2335
62
+ Stropharia aeruginosa 0.9800 0.9846 0.9823 2335
63
+ Cladonia fimbriata 0.9514 0.9717 0.9614 2335
64
+ Suillus grevillei 0.9241 0.3597 0.5179 2335
65
+ Apioperdon pyriforme 0.9025 0.9512 0.9262 2335
66
+ Cerioporus squamosus 0.9422 0.9696 0.9557 2335
67
+ Leccinum scabrum 0.7117 0.9199 0.8025 2335
68
+ Rhytisma acerinum 0.9996 0.9927 0.9961 2335
69
+ Hypholoma lateritium 0.7966 0.9126 0.8506 2334
70
+ Flammulina velutipes 0.8807 0.8985 0.8895 2334
71
+ Tricholomopsis rutilans 0.9347 0.7910 0.8569 2335
72
+ Coprinopsis atramentaria 0.9165 0.9208 0.9186 2335
73
+ Trametes versicolor 0.7984 0.8972 0.8449 2335
74
+ Graphis scripta 0.9726 0.9863 0.9794 2335
75
+ Ganoderma applanatum 0.9201 0.9572 0.9383 2335
76
+ Phellinus tremulae 0.9042 0.8530 0.8779 2334
77
+ Peltigera aphthosa 0.9787 0.9816 0.9801 2335
78
+ Parmelia sulcata 0.8846 0.9263 0.9050 2334
79
+ Fomitopsis betulina 0.8569 0.9640 0.9073 2335
80
+ Pleurotus pulmonarius 0.8792 0.8818 0.8805 2335
81
+ Fomitopsis pinicola 0.9419 0.9649 0.9532 2335
82
+ Daedaleopsis confragosa 0.7362 0.8642 0.7951 2335
83
+ Hericium coralloides 0.9889 0.9880 0.9884 2335
84
+ Trametes hirsuta 0.8508 0.8206 0.8354 2335
85
+ Coprinellus disseminatus 0.9320 0.9447 0.9383 2334
86
+ Kuehneromyces mutabilis 0.7370 0.9263 0.8209 2335
87
+ Pleurotus ostreatus 0.6688 0.8951 0.7656 2335
88
+ Phlebia radiata 0.9595 0.9448 0.9521 2335
89
+ Boletus reticulatus 0.9420 0.0835 0.1534 2335
90
+ Phallus impudicus 0.9978 0.9559 0.9764 2334
91
+ Macrolepiota procera 0.9689 0.9880 0.9784 2335
92
+ Fomes fomentarius 0.8935 0.9314 0.9121 2334
93
+ Suillus granulatus 0.4304 0.9233 0.5871 2335
94
+ Gyromitra esculenta 0.9349 0.9473 0.9411 2335
95
+ Xanthoria parietina 0.9507 0.9666 0.9586 2335
96
+ Nectria cinnabarina 0.9836 0.9739 0.9787 2335
97
+ Sarcomyxa serotina 0.9569 0.3135 0.4723 2335
98
+ Inonotus obliquus 0.9537 0.9966 0.9747 2335
99
+ Panellus stipticus 0.8801 0.7919 0.8336 2335
100
+ Hypogymnia physodes 0.8664 0.9413 0.9023 2335
101
+ Hygrophoropsis aurantiaca 0.9111 0.8963 0.9037 2334
102
+ Cladonia rangiferina 0.9403 0.8839 0.9113 2335
103
+ Platismatia glauca 0.9447 0.9507 0.9477 2334
104
+ Calycina citrina 0.9764 0.9940 0.9851 2335
105
+ Cladonia stellaris 0.9127 0.9623 0.9368 2335
106
+ Amanita citrina 0.9165 0.9829 0.9485 2335
107
+ Lepista nuda 0.9643 0.9722 0.9682 2335
108
+ Gyromitra gigas 0.9669 0.9623 0.9646 2335
109
+ Crucibulum laeve 0.9153 0.9576 0.9360 2335
110
+ Daedaleopsis tricolor 0.9066 0.8026 0.8514 2335
111
+ Stereum hirsutum 0.8651 0.8540 0.8595 2335
112
+ Paxillus involutus 0.7272 0.9272 0.8151 2335
113
+ Lactarius turpis 0.9264 0.8788 0.9020 2335
114
+ Chlorociboria aeruginascens 0.9996 0.9893 0.9944 2335
115
+ Chondrostereum purpureum 0.9384 0.8878 0.9124 2335
116
+ Phaeophyscia orbicularis 0.9021 0.7734 0.8328 2335
117
+ Peltigera praetextata 0.9816 0.9572 0.9692 2335
118
 
119
+ accuracy 0.8854 233480
120
+ macro avg 0.8961 0.8854 0.8797 233480
121
+ weighted avg 0.8961 0.8854 0.8797 233480
122
  ```