dcarpintero/fastai-interstellar-class

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:

Class Model

This version fastai-interstellar-class

Classifies an image under an astronomy class:

• asteroid
• comet
• galaxy
• nebulae
• planet
• quasar in space
• star cluster

Accuracy: 84%

---

Object Model

Available at 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%