Returns lemon quality given an image.

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

Classification report:

                  precision    recall  f1-score   support

    good_quality     1.0000    1.0000    1.0000       450
empty_background     1.0000    1.0000    1.0000       450
     bad_quality     1.0000    1.0000    1.0000       450

        accuracy                         1.0000      1350
       macro avg     1.0000    1.0000    1.0000      1350
    weighted avg     1.0000    1.0000    1.0000      1350

Downloads last month: 24

Safetensors

Model size

85.8M params

Tensor type

F32

Inference Examples

Image Classification

This model does not have enough activity to be deployed to Inference API (serverless) yet. Increase its social visibility and check back later, or deploy to Inference Endpoints (dedicated) instead.

Model tree for dima806/lemon_quality_image_detection

Base model

google/vit-base-patch16-224-in21k

Finetuned

(1724)

this model