Astronomical Classifier
Classify images of astronomical objects such as galaxies, nebulae, comets, asteroids, quasars, and star clusters.
Built for the fast.ai Practical Deep Learning Course by:
- creating a custom dataset (less than 150 images per label) using Bing search API;
- augmenting the dataset; and,
- fine tuning ResNet50 (1 + 3 epochs) in paperspace.com.
Try at: https://huggingface.co/spaces/dcarpintero/fastai-interstellar
Two versions of the model are provided:
Object Model
This version fastai-interstellar-object
Recognizes a (limited) set of specific astronomical objects:
m31 andromeda
m33 triangulum
m81 bode
m82 cigar
ngc 1300
m104 sombrero
m51 whirlpool
m42 orion nebula
m17 omega nebula
m45 pleiades star cluster
Accuracy: 94.1%
Class Model
Available at fastai-interstellar-class
Classifies an image under an astronomy class:
asteroid
comet
galaxy
nebulae
planet
quasar in space
star cluster
Accuracy: 84%
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.