Edit model card

Returns flower group given an image.

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

image/png

Classification report:

                   precision    recall  f1-score   support

         tarragon     0.0000    0.0000    0.0000       247
   blanketflowers     0.9868    0.9109    0.9474       247
       coralbells     0.8791    0.9717    0.9231       247
           tulips     0.9741    0.9150    0.9436       247
        daffodils     0.8719    0.9919    0.9280       247
             peas     0.8972    0.9190    0.9080       247
           garlic     0.0000    0.0000    0.0000       247
         sunroots     1.0000    0.0486    0.0927       247
         milkweed     0.8509    0.9474    0.8966       247
           celery     0.0000    0.0000    0.0000       247
             dill     0.4819    0.9717    0.6443       247
            phlox     0.9137    0.9433    0.9283       247
          peonies     0.5545    0.9879    0.7103       247
      coneflowers     0.9679    0.9757    0.9718       247
            beets     0.7526    0.8745    0.8090       247
            beans     0.8824    0.9756    0.9266       246
           onions     0.3012    0.9231    0.4542       247
      bellflowers     0.9865    0.8907    0.9362       247
      delphiniums     0.9955    0.8866    0.9379       247
        oleanders     0.9875    0.9595    0.9733       247
     roseofsharon     0.9727    0.4350    0.6011       246
      cantaloupes     0.9329    0.6194    0.7445       247
      deadnettles     0.9534    0.9109    0.9317       247
        viburnums     0.5501    0.8664    0.6730       247
         dianthus     0.8298    0.9512    0.8864       246
          peaches     0.8902    0.5911    0.7105       247
            aloes     0.7724    0.9757    0.8623       247
          parsley     0.3561    0.9717    0.5212       247
        penstemon     0.9782    0.9106    0.9432       246
            thyme     0.6685    0.9879    0.7974       247
           citrus     0.8479    0.9028    0.8745       247
  bleeding-hearts     0.9679    0.9757    0.9718       247
         dogwoods     0.5442    0.9231    0.6847       247
black-eyed-susans     0.5501    0.9555    0.6982       247
         petunias     0.9790    0.9472    0.9628       246
          jujubes     0.0000    0.0000    0.0000       247
      arborvitaes     0.8880    0.8664    0.8770       247
           lilies     0.9783    0.9109    0.9434       247
          crinums     0.7704    0.8421    0.8046       247
         catmints     0.6334    0.8745    0.7347       247
          astilbe     0.9597    0.9636    0.9616       247
    beautyberries     0.7500    0.8988    0.8177       247
         beebalms     0.8484    0.9514    0.8969       247
        foxgloves     0.9713    0.9595    0.9654       247
        gladiolus     0.9048    0.9231    0.9138       247
            plums     0.8571    0.4615    0.6000       247
            vitis     1.0000    0.5466    0.7068       247
        ninebarks     1.0000    0.0445    0.0853       247
         lettuces     0.7921    0.8947    0.8403       247
          poppies     0.9679    0.9757    0.9718       247
       smoketrees     0.9202    0.8866    0.9031       247
           irises     1.0000    0.9960    0.9980       247
         cilantro     0.9600    0.0972    0.1765       247
       artichokes     1.0000    0.7895    0.8824       247
        lambsears     0.6519    0.7764    0.7087       246
      butterworts     0.9286    0.2105    0.3432       247
      babysbreath     1.0000    0.1700    0.2907       247
        cucurbits     0.5658    0.9959    0.7216       246
        plumerias     0.8051    0.8902    0.8456       246
          liatris     0.9720    0.8455    0.9043       246
          carrots     0.6364    0.5407    0.5846       246
    crepe-myrtles     0.9710    0.9474    0.9590       247
          oregano     0.6372    0.2927    0.4011       246
             ilex     0.5610    0.9676    0.7103       247
  butterflybushes     0.9726    0.8623    0.9142       247
             sage     0.4910    0.4413    0.4648       247
         baptisia     0.9744    0.7692    0.8597       247
      sempervivum     0.9910    0.8943    0.9402       246
        asparagus     0.9610    0.3008    0.4582       246
         radishes     0.5153    0.7490    0.6106       247
         parsnips     1.0000    0.1174    0.2101       247
         hibiscus     0.4605    0.9715    0.6248       246
    rhododendrons     0.8918    0.9676    0.9282       247
         potatoes     1.0000    0.4130    0.5845       247
       hydrangeas     0.9504    0.9350    0.9426       246
       swisschard     0.8154    0.9878    0.8934       246
           cannas     0.9360    0.9474    0.9416       247
        brassicas     0.6437    0.8740    0.7414       246
            rubus     0.8631    0.8421    0.8525       247
       columbines     0.9717    0.9717    0.9717       247
       echeverias     0.6384    0.9150    0.7521       247
             okra     0.9901    0.8138    0.8933       247
         aeoniums     0.5124    0.9190    0.6580       247
          yarrows     0.7126    0.9636    0.8193       247
            roses     0.9880    0.9960    0.9919       247
            basil     0.6419    0.9433    0.7639       247
         spiraeas     0.5897    0.9717    0.7339       247
        caladiums     0.7804    0.9352    0.8508       247
          spinach     0.8947    0.2753    0.4211       247
        wisterias     0.9609    0.8947    0.9266       247
         cherries     1.0000    0.1862    0.3140       247
         marjoram     1.0000    0.3927    0.5640       247
        hyacinths     0.9711    0.9514    0.9611       247
         rhubarbs     0.9651    0.8947    0.9286       247
        tickseeds     0.8588    0.8866    0.8725       247
        perovskia     0.7869    0.5830    0.6698       247
           crocus     0.9789    0.9431    0.9607       246
            mints     0.6088    0.9514    0.7425       247
 heavenly-bamboos     0.9493    0.8340    0.8879       247
           agaves     0.9025    0.8623    0.8820       247
            pears     0.3087    0.4575    0.3687       247
         dudleyas     0.8291    0.5304    0.6469       247
     pachypodiums     0.8820    0.6356    0.7388       247
      mockoranges     0.9958    0.9676    0.9815       247
           asters     0.9957    0.9512    0.9730       246
        geraniums     0.9750    0.9474    0.9610       247
     mammillarias     0.9447    0.9715    0.9579       246
        cucumbers     1.0000    0.6235    0.7681       247
        veronicas     0.9368    0.9595    0.9480       247
          turnips     0.0000    0.0000    0.0000       247
          peppers     0.8053    0.9919    0.8889       246
  hardyhibiscuses     1.0000    0.4593    0.6295       246
  morning-glories     0.8316    0.9595    0.8910       247
        gardenias     0.9954    0.8785    0.9333       247
            ribes     0.9837    0.7358    0.8419       246
        loniceras     0.9540    0.9231    0.9383       247
        eggplants     0.9837    0.9798    0.9817       247
           hostas     0.8167    0.9919    0.8958       247
    chlorophytums     0.9709    0.6761    0.7971       247
           chives     0.7029    0.9676    0.8143       247
         tomatoes     0.6619    0.9352    0.7752       247
           lilacs     1.0000    0.9595    0.9793       247
            leeks     0.0000    0.0000    0.0000       246
    shastadaisies     0.9592    0.9514    0.9553       247
         apricots     1.0000    0.5830    0.7366       247
           apples     0.4027    0.9636    0.5680       247
     strawberries     0.8897    0.9798    0.9326       247
          salvias     0.4479    0.9393    0.6065       247
           sedums     0.7639    0.9472    0.8457       246
             corn     0.9129    0.8907    0.9016       247
        daylilies     1.0000    0.9960    0.9980       247
             figs     0.9711    0.9553    0.9631       246
          dahlias     0.9757    0.9757    0.9757       247
    sweetpotatoes     0.7183    0.9393    0.8140       247

         accuracy                         0.7785     33072
        macro avg     0.8044    0.7785    0.7529     33072
     weighted avg     0.8044    0.7785    0.7528     33072
Downloads last month
15
Safetensors
Model size
85.9M params
Tensor type
F32
·