Edit model card

Returns mushroom type given an image.

See https://www.kaggle.com/code/dima806/mushroom-types-image-detection-vit for more details.

Classification report:

                          precision    recall  f1-score   support

         mosaic_puffball     1.0000    1.0000    1.0000         7
          scarlet_elfcup     1.0000    1.0000    1.0000         7
         splendid_waxcap     1.0000    0.4286    0.6000         7
          tawny_grisette     0.8750    1.0000    0.9333         7
          jubilee_waxcap     1.0000    1.0000    1.0000         6
      king_alfreds_cakes     1.0000    0.8333    0.9091         6
            heath_waxcap     0.7500    1.0000    0.8571         6
          silky_rosegill     1.0000    1.0000    1.0000         6
           golden_waxcap     0.4286    1.0000    0.6000         6
          macro_mushroom     1.0000    0.8571    0.9231         7
    spectacular_rustgill     0.7500    0.8571    0.8000         7
             pink_waxcap     1.0000    1.0000    1.0000         6
      brown_birch_bolete     0.8333    0.8333    0.8333         6
     scaly_wood_mushroom     1.0000    1.0000    1.0000         6
               stinkhorn     0.8571    1.0000    0.9231         6
  blackening_brittlegill     1.0000    0.7143    0.8333         7
               penny_bun     0.8571    1.0000    0.9231         6
    chicken_of_the_woods     1.0000    1.0000    1.0000         7
           common_bonnet     1.0000    0.7143    0.8333         7
         common_rustgill     1.0000    0.8333    0.9091         6
         hedgehog_fungus     1.0000    0.8333    0.9091         6
         shaggy_scalycap     1.0000    0.8333    0.9091         6
          dyers_mazegill     0.8571    1.0000    0.9231         6
              earthballs     1.0000    1.0000    1.0000         7
      purple_brittlegill     1.0000    0.8333    0.9091         6
           smoky_bracket     0.7143    0.7143    0.7143         7
            elfin_saddle     1.0000    1.0000    1.0000         6
          shaggy_bracket     0.7778    1.0000    0.8750         7
greencracked_brittlegill     1.0000    0.6667    0.8000         6
            sulphur_tuft     1.0000    1.0000    1.0000         6
          warted_amanita     1.0000    0.7143    0.8333         7
           white_domecap     0.7778    1.0000    0.8750         7
      winter_chanterelle     1.0000    1.0000    1.0000         7
             grey_knight     1.0000    0.8571    0.9231         7
             pale_oyster     1.0000    0.5714    0.7273         7
         medusa_mushroom     0.6667    0.8571    0.7500         7
      spotted_toughshank     1.0000    1.0000    1.0000         7
           dog_stinkhorn     1.0000    0.8333    0.9091         6
        stubble_rosegill     1.0000    0.6667    0.8000         6
                truffles     1.0000    1.0000    1.0000         6
              panthercap     0.8000    0.6667    0.7273         6
       vermillion_waxcap     1.0000    1.0000    1.0000         7
               ascot_hat     0.8571    1.0000    0.9231         6
          birch_polypore     1.0000    0.5000    0.6667         6
            common_morel     0.7778    1.0000    0.8750         7
          shaggy_parasol     1.0000    0.6667    0.8000         6
             turkey_tail     0.6667    1.0000    0.8000         6
             the_blusher     0.6250    0.8333    0.7143         6
                deathcap     0.3333    1.0000    0.5000         7
         chestnut_bolete     1.0000    0.7143    0.8333         7
    grey_spotted_amanita     1.0000    0.8571    0.9231         7
         slender_parasol     1.0000    0.8571    0.9231         7
          horn_of_plenty     1.0000    1.0000    1.0000         7
           magpie_inkcap     1.0000    0.8333    0.9091         6
            fools_funnel     0.8333    0.8333    0.8333         6
     orange_birch_bolete     1.0000    1.0000    1.0000         6
          scarlet_waxcap     0.5714    0.6667    0.6154         6
          yellow_stainer     1.0000    0.6667    0.8000         6
          field_mushroom     1.0000    0.8333    0.9091         6
         fragrant_funnel     0.8333    0.8333    0.8333         6
         spring_fieldcap     0.8333    0.7143    0.7692         7
           bronze_bolete     1.0000    0.4286    0.6000         7
         orange_grisette     1.0000    0.8571    0.9231         7
                 parasol     0.8333    0.7143    0.7692         7
         trooping_funnel     1.0000    0.7143    0.8333         7
      beechwood_sickener     1.0000    0.6667    0.8000         6
             rosy_bonnet     0.8333    0.8333    0.8333         6
          dusky_puffball     1.0000    1.0000    1.0000         7
              the_miller     0.7000    1.0000    0.8235         7
            white_saddle     1.0000    1.0000    1.0000         7
    old_man_of_the_woods     1.0000    1.0000    1.0000         6
            crimped_gill     1.0000    0.8333    0.9091         6
        blushing_rosette     1.0000    1.0000    1.0000         6
             pine_bolete     1.0000    1.0000    1.0000         6
           brown_rollrim     1.0000    0.8333    0.9091         6
           deadly_webcap     1.0000    1.0000    1.0000         7
           devils_bolete     1.0000    1.0000    1.0000         6
 scarlet_caterpillarclub     1.0000    1.0000    1.0000         7
     red_cracking_bolete     1.0000    1.0000    1.0000         6
       false_chanterelle     1.0000    0.8333    0.9091         6
         woodland_inkcap     0.6667    0.8571    0.7500         7
            cucumber_cap     1.0000    0.8571    0.9231         7
 leccinum_albostipitatum     1.0000    1.0000    1.0000         6
  fairy_ring_champignons     0.8333    0.8333    0.8333         6
          rooting_bolete     0.7500    1.0000    0.8571         6
             wood_blewit     0.7500    1.0000    0.8571         6
            lilac_bonnet     0.8333    0.8333    0.8333         6
              butter_cap     1.0000    1.0000    1.0000         7
            black_bulgar     1.0000    1.0000    1.0000         7
          giant_puffball     0.8571    1.0000    0.9231         6
          false_deathcap     0.0000    0.0000    0.0000         6
          white_fibrecap     1.0000    1.0000    1.0000         6
            velvet_shank     1.0000    0.8571    0.9231         7
           slippery_jack     0.5556    0.8333    0.6667         6
        white_dapperling     0.6667    0.8571    0.7500         7
           parrot_waxcap     1.0000    0.8333    0.9091         6
          wrinkled_peach     0.8571    1.0000    0.9231         6
       silverleaf_fungus     1.0000    1.0000    1.0000         7
         amanita_gemmata     1.0000    1.0000    1.0000         6
     stinking_dapperling     1.0000    0.8333    0.9091         6
       plums_and_custard     1.0000    0.6667    0.8000         6
          peppery_bolete     0.8000    0.6667    0.7273         6
     terracotta_hedgehog     0.8333    0.8333    0.8333         6
      egghead_mottlegill     1.0000    1.0000    1.0000         6
         bearded_milkcap     1.0000    0.8333    0.9091         6
           inky_mushroom     1.0000    0.5000    0.6667         6
            larch_bolete     0.8571    0.8571    0.8571         7
        porcelain_fungus     0.8571    1.0000    0.9231         6
             jelly_tooth     1.0000    1.0000    1.0000         6
       scarletina_bolete     0.5000    1.0000    0.6667         6
      yellow_foot_waxcap     1.0000    1.0000    1.0000         6
              the_prince     1.0000    0.5000    0.6667         6
          aniseed_funnel     1.0000    0.8333    0.9091         6
   white_false_death_cap     0.5000    0.8333    0.6250         6
   false_saffron_milkcap     1.0000    0.8333    0.9091         6
yellow_swamp_brittlegill     1.0000    0.8333    0.9091         6
          semifree_morel     1.0000    1.0000    1.0000         7
           bitter_bolete     1.0000    0.7143    0.8333         7
         almond_mushroom     1.0000    1.0000    1.0000         6
           shaggy_inkcap     0.8750    1.0000    0.9333         7
  blushing_wood_mushroom     1.0000    0.6667    0.8000         6
         common_puffball     1.0000    1.0000    1.0000         6
            funeral_bell     0.7500    1.0000    0.8571         6
              bay_bolete     1.0000    0.8333    0.9091         6
       blackening_waxcap     1.0000    0.5714    0.7273         7
             liberty_cap     0.6000    1.0000    0.7500         6
            snowy_waxcap     0.6667    1.0000    0.8000         6
              the_goblet     1.0000    1.0000    1.0000         7
             deer_shield     1.0000    1.0000    1.0000         7
     freckled_dapperling     0.6667    1.0000    0.8000         6
            slimy_waxcap     0.6667    1.0000    0.8000         6
           common_inkcap     0.7778    1.0000    0.8750         7
    amethyst_chanterelle     0.8750    1.0000    0.9333         7
        cedarwood_waxcap     0.7143    0.8333    0.7692         6
            honey_fungus     1.0000    0.8571    0.9231         7
         bruising_webcap     1.0000    0.4286    0.6000         7
          stump_puffball     0.8571    1.0000    0.9231         6
            giant_funnel     0.8333    0.8333    0.8333         6
       tuberous_polypore     1.0000    0.6667    0.8000         6
              poison_pie     0.8571    0.8571    0.8571         7
           curry_milkcap     1.0000    1.0000    1.0000         6
       amethyst_deceiver     1.0000    1.0000    1.0000         7
          golden_bootleg     1.0000    0.7143    0.8333         7
       clustered_domecap     1.0000    0.6667    0.8000         6
       ochre_brittlegill     0.7143    0.7143    0.7143         7
     blackening_polypore     1.0000    0.8333    0.9091         6
            suede_bolete     1.0000    1.0000    1.0000         7
          horse_mushroom     0.5455    1.0000    0.7059         6
    geranium_brittlegill     0.6667    1.0000    0.8000         6
     st_georges_mushroom     1.0000    0.8333    0.9091         6
        destroying_angel     0.0000    0.0000    0.0000         6
            field_blewit     1.0000    0.5714    0.7273         7
        cinnamon_bracket     1.0000    1.0000    1.0000         6
              lions_mane     1.0000    0.8333    0.9091         6
      orange_peel_fungus     1.0000    1.0000    1.0000         6
             chanterelle     0.8750    1.0000    0.9333         7
            the_sickener     0.8571    1.0000    0.9231         6
          birch_woodwart     0.8571    1.0000    0.9231         6
       pavement_mushroom     0.7500    1.0000    0.8571         6
             false_morel     1.0000    1.0000    1.0000         7
              oak_bolete     1.0000    0.8333    0.9091         6
         poplar_fieldcap     1.0000    0.5000    0.6667         6
              jelly_ears     1.0000    1.0000    1.0000         6
           summer_bolete     0.6250    0.8333    0.7143         6
     frosted_chanterelle     0.5714    0.6667    0.6154         6
                   morel     1.0000    0.8333    0.9091         6
            the_deceiver     1.0000    0.8571    0.9231         7
               splitgill     0.8571    1.0000    0.9231         6
             ruby_bolete     0.8571    0.8571    0.8571         7
            sepia_bolete     1.0000    0.5714    0.7273         7
           bovine_bolete     0.8750    1.0000    0.9333         7
              fly_agaric     1.0000    1.0000    1.0000         7
           thimble_morel     0.8571    1.0000    0.9231         6
             black_morel     0.8333    0.8333    0.8333         6
             poplar_bell     1.0000    1.0000    1.0000         6
          fleecy_milkcap     0.7778    1.0000    0.8750         7
         golden_scalycap     0.7500    1.0000    0.8571         6
        yellow_stagshorn     1.0000    1.0000    1.0000         6
            oak_polypore     1.0000    0.8333    0.9091         6
           weeping_widow     0.7500    0.8571    0.8000         7
           meadow_waxcap     0.8750    1.0000    0.9333         7
          clouded_agaric     0.7500    0.8571    0.8000         7
          woolly_milkcap     0.8750    1.0000    0.9333         7
      snakeskin_grisette     1.0000    0.8333    0.9091         6
     hairy_curtain_crust     0.8750    1.0000    0.9333         7
            lurid_bolete     1.0000    0.6667    0.8000         6
           wood_mushroom     0.8571    0.8571    0.8571         7
           dryads_saddle     0.8750    1.0000    0.9333         7
       sheathed_woodtuft     1.0000    0.8571    0.9231         7
           orange_bolete     0.6667    1.0000    0.8000         6
          lilac_fibrecap     1.0000    0.8571    0.9231         7
      cauliflower_fungus     1.0000    1.0000    1.0000         7
         saffron_milkcap     0.7500    0.5000    0.6000         6
         pestle_puffball     1.0000    0.8571    0.9231         7
      red_belted_bracket     1.0000    1.0000    1.0000         6
        beefsteak_fungus     1.0000    1.0000    1.0000         7
            oak_mazegill     1.0000    0.4286    0.6000         7
       glistening_inkcap     0.8571    0.8571    0.8571         7
            tripe_fungus     1.0000    0.6667    0.8000         6
        blushing_bracket     0.7143    0.7143    0.7143         7
         deadly_fibrecap     0.8571    1.0000    0.9231         6
                root_rot     0.5556    0.8333    0.6667         6
     powdery_brittlegill     1.0000    1.0000    1.0000         6
               grisettes     0.6667    0.6667    0.6667         6
         charcoal_burner     0.8333    0.7143    0.7692         7
           rooting_shank     1.0000    1.0000    1.0000         6
        hen_of_the_woods     0.8571    1.0000    0.9231         6
          crimson_waxcap     1.0000    1.0000    1.0000         6
       fenugreek_milkcap     1.0000    1.0000    1.0000         7
         oyster_mushroom     0.6667    1.0000    0.8000         6
          blue_roundhead     0.8571    1.0000    0.9231         6
             hoof_fungus     0.7500    1.0000    0.8571         6
     bitter_beech_bolete     1.0000    0.5714    0.7273         7
            tawny_funnel     1.0000    1.0000    1.0000         6
    yellow_false_truffle     1.0000    1.0000    1.0000         6

                accuracy                         0.8699      1376
               macro avg     0.8933    0.8701    0.8670      1376
            weighted avg     0.8949    0.8699    0.8676      1376
Downloads last month
39
Safetensors
Model size
86M params
Tensor type
F32
·