Returns coffee roast type given bean image.

See https://www.kaggle.com/code/dima806/roasted-coffee-bean-image-detection-vit for more details.

Classification report:

              precision    recall  f1-score   support

        Dark     1.0000    1.0000    1.0000       160
       Light     1.0000    1.0000    1.0000       160
       Green     1.0000    1.0000    1.0000       160
      Medium     1.0000    1.0000    1.0000       160

    accuracy                         1.0000       640
   macro avg     1.0000    1.0000    1.0000       640
weighted avg     1.0000    1.0000    1.0000       640

Downloads last month: 11

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/coffee_bean_roast_image_detection

Base model

google/vit-base-patch16-224-in21k

Finetuned

(1693)

this model