dcarpintero
commited on
Commit
•
4f801af
1
Parent(s):
e6e25c3
Update README.md
Browse files
README.md
CHANGED
@@ -1,3 +1,57 @@
|
|
1 |
---
|
2 |
license: apache-2.0
|
|
|
|
|
|
|
|
|
|
|
|
|
3 |
---
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
---
|
2 |
license: apache-2.0
|
3 |
+
metrics:
|
4 |
+
- accuracy
|
5 |
+
pipeline_tag: image-classification
|
6 |
+
tags:
|
7 |
+
- fastai
|
8 |
+
- astronomy
|
9 |
---
|
10 |
+
|
11 |
+
## Interstellar Classifier
|
12 |
+
|
13 |
+
Classify images of interstellar objects such as galaxies, nebulae, comets, asteroids, quasars, and star clusters.
|
14 |
+
|
15 |
+
Built for the fast.ai [Practical Deep Learning Course](https://course.fast.ai/) by:
|
16 |
+
- 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);
|
17 |
+
- augmenting the dataset; and,
|
18 |
+
- fine tuning ResNet50 (1 + 3 epochs) in [paperspace.com](https://www.paperspace.com/).
|
19 |
+
|
20 |
+
Try at: https://huggingface.co/spaces/dcarpintero/fastai-interstellar
|
21 |
+
|
22 |
+
Two versions of the model are provided:
|
23 |
+
|
24 |
+
### Object Model
|
25 |
+
|
26 |
+
*This version [fastai-interstellar-object](https://huggingface.co/dcarpintero/fastai-interstellar-object/tree/main)*
|
27 |
+
|
28 |
+
Recognizes a (limited) set of **specific** interstellar objects:
|
29 |
+
- `m31 andromeda`
|
30 |
+
- `m33 triangulum`
|
31 |
+
- `m81 bode`
|
32 |
+
- `m82 cigar`
|
33 |
+
- `ngc 1300`
|
34 |
+
- `m104 sombrero`
|
35 |
+
- `m51 whirlpool`
|
36 |
+
- `m42 orion nebula`
|
37 |
+
- `m17 omega nebula`
|
38 |
+
- `m45 pleiades star cluster`
|
39 |
+
|
40 |
+
Accuracy: 94.1%
|
41 |
+
|
42 |
+
----
|
43 |
+
|
44 |
+
### Class Model
|
45 |
+
|
46 |
+
*Available at [fastai-interstellar-class](https://huggingface.co/dcarpintero/fastai-interstellar-class/tree/main)*
|
47 |
+
|
48 |
+
Classifies an image under an astronomy class:
|
49 |
+
- `asteroid`
|
50 |
+
- `comet`
|
51 |
+
- `galaxy`
|
52 |
+
- `nebulae`
|
53 |
+
- `planet`
|
54 |
+
- `quasar in space`
|
55 |
+
- `star cluster`
|
56 |
+
|
57 |
+
Accuracy: 84%
|