Downloads · 30 days
48
4% of all-time downloads
galaxy_type_image_detection
dima806/galaxy_type_image_detection
galaxy_type_image_detection is a image classification model from dima806. Use it when you need a label for an image. It is set up for transformers. The card lists the license as apache-2.0.
Achieved 78% weighted accuracy for classification between 3 common galaxy types (S, E, Sb).
All-time downloads
1.2K
Public
Parameters
85.8M
3.4 GB on disk
Likes
1
Public
Repo makeup
Click a slice to open those files.
.pt2.1 GB · 55%
At a glance
- Task
- Image Classification
- Library
- transformers
- License
- apache-2.0
- Model type
- vit
- Access
- Public
- Created
- Sep 4, 2023
- Updated
- Oct 19, 2024
- SHA
- 0e679339
Base models
Model card
From the Hugging Face model README
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
- Task
- Image Classification
- Library
- transformers
- Type
- vit
- License
- apache-2.0
- Created
- Sep 4, 2023
- Updated
- Oct 19, 2024