dcarpintero commited on
Commit
4f801af
1 Parent(s): e6e25c3

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +54 -0
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%