Achieved 78% weighted accuracy for classification between 3 common galaxy types (S, E, Sb).

See my Kaggle notebook for more details.

Classification report:

              precision    recall  f1-score   support

           E     0.7656    0.8848    0.8209     13592
           S     0.7526    0.6685    0.7081     13591
          SB     0.8262    0.7900    0.8077     13591

    accuracy                         0.7811     40774
   macro avg     0.7815    0.7811    0.7789     40774
weighted avg     0.7815    0.7811    0.7789     40774

Downloads last month: 7

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

Base model

google/vit-base-patch16-224-in21k

Finetuned

(1704)

this model