fastai astronomy image-classification license:apache-2.0 region:us
---
license: apache-2.0
metrics:
- accuracy
pipeline_tag: image-classification
tags:
- fastai
- astronomy
---

## 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](https://course.fast.ai/) by:
- creating a custom dataset (less than 150 images per label) using [Bing search API](https://www.microsoft.com/en-us/bing/apis/bing-web-search-api);
- augmenting the dataset; and,
- fine tuning ResNet50 (1 + 3 epochs) in [paperspace.com](https://www.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](https://huggingface.co/dcarpintero/fastai-interstellar-object/tree/main)*

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](https://huggingface.co/dcarpintero/fastai-interstellar-class/tree/main)*

Classifies an image under an astronomy class:
- `asteroid`
- `comet` 
- `galaxy`
- `nebulae`
- `planet`
- `quasar in space` 
- `star cluster`
 
Accuracy: 84%