Edit model card

Returns flower type given an image.

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

image/png

Classification report:

                  precision    recall  f1-score   support

            rose     0.9951    0.9737    0.9843       419
         astilbe     0.9952    0.9905    0.9928       419
       carnation     0.9627    0.9857    0.9741       419
           tulip     0.9929    1.0000    0.9964       420
      water_lily     1.0000    0.9905    0.9952       419
      bellflower     0.9811    0.9905    0.9857       419
       coreopsis     0.9881    0.9881    0.9881       419
    common_daisy     0.9858    0.9928    0.9893       419
            iris     0.9953    1.0000    0.9976       420
       dandelion     0.9905    1.0000    0.9952       419
       sunflower     0.9976    0.9976    0.9976       419
california_poppy     0.9951    0.9761    0.9855       419
black_eyed_susan     0.9882    1.0000    0.9941       419
       calendula     0.9854    0.9667    0.9760       420

        accuracy                         0.9894      5869
       macro avg     0.9895    0.9894    0.9894      5869
    weighted avg     0.9895    0.9894    0.9894      5869
Downloads last month
2
Safetensors
Model size
85.8M params
Tensor type
F32
·