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