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: 8

Safetensors

Model size

85.8M params

Tensor type

F32

Inference Providers NEW

Image Classification

This model isn't deployed by any Inference Provider. 🙋 Ask for provider support

Model tree for dima806/galaxy_type_image_detection

Base model

google/vit-base-patch16-224-in21k

Finetuned

(2009)

this model