File size: 1,420 Bytes
853e78e
 
f6d2e9b
 
 
 
 
 
853e78e
f6d2e9b
b117885
f6d2e9b
b117885
f6d2e9b
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
b117885
f6d2e9b
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
---
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:

### Class Model

*This version [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%

----

### Object Model 

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