Upload with huggingface_hub

README.md

@@ -6,27 +6,82 @@ tags:
 - text-to-image
 ---
 
-#Stellar Diffusion
+# Stellar Diffusion
+
+Stellar Diffusion v0.1 vs Base Stable Diffusion v1.5
+<img src="images/galaxy_std.png" width="300" height="300"> <img src="images/galaxy_sd.png" width="300" height="300">
+```
+prompt = A hubble photograph of a galaxy
+seed = 42
+size = 512x512
+```
+
 
 Version: 0.1
 
 Stable Diffusion 1.5 finetuned on high quality processed space imagery.
 
 ---
+# Example Results
+
+<img src="images/barredspiral.png" width="200" height="200"> 
+```
+prompt = A barred spiral galaxy
+seed = 44
+size = 512x512
+```
+
+<img src="images/brightconstellation.png" width="200" height="200"> 
+```
+prompt = bright, constellation, hubble, clouds
+seed = 42
+size = 512x512
+```
 
-#Example Prompt
+<img src="images/nebula.png" width="200" height="200"> 
+```
+prompt = a hubble photograph of a nebula
+seed = 42
+size = 512x512
+```
 
+<img src="images/ngc7714.png" width="200" height="200"> 
 ```
-A hubble photograph of a galaxy
+prompt = ngc 7714
+seed = 42
+size = 512x512
 ```
 
-##Suggested parameters
+---
+
+# Suggested parameters
 
 512x512
 
 ---
 
-#Python Usage
+# Reconized Tags
+
+All reconized tags can be found in the tags.txt file. They are generated from the annotated descriptions of the photograph.
+Current Dataset is small and is poor at generating exact celestial bodies, but is better at generating common generic bodies like nebula, galaxies etc.
+
+### Partial support for scientific celestial body tags as follows:
+
+NGC - New General Catalogue of Nebulae and Clusters of Stars
+
+M / Messier - A set of 110 astronomical objects catalogued by the French astronomer Charles Messier
+
+UGC – (catalog) Uppsala General Catalogue, a catalog of galaxies
+
+### Partial support for the following classification methods as follows:
+
+By recording instrument/spacecraft (ex. Voyager, Hubble)
+
+By Color
+
+By Celestial Body type
+
+# Python Usage
 
 ```py
 from diffusers import StableDiffusionPipeline
@@ -44,11 +99,43 @@ image.save("hubble_galaxy.png")
 
 ---
 
-#Dataset
+# Dataset and Credits
+
+### Model
+
+Rex Wang (me!)
+
+RunwayML for their SD 1.5
+
+### Compute
+
+Coreweave - 2x A40s
+
+### Dataset
 
 91 of the 100 images from https://esahubble.org/ Top 100 Hubble Images
 ESA/Hubble
 
 ---
 
+# Planned
+
+Expansion of dataset to include:
+
+Solar system
+
+Asteroids
+
+More star types
+
+Black holes
+
+Exo planets
+
+More data in general
+
+---
+
+# Version History
 
+V0.1 - 91 image dataset

tags.txt

@@ -0,0 +1,281 @@
+hubble: 81
+space: 65
+red: 63
+hubble space telescope: 62
+star: 61
+nasa: 56
+esa: 56
+galaxy: 53
+gas: 37
+constellation: 36
+dust: 35
+galaxies: 32
+spiral: 29
+bright: 29
+earth: 28
+cloud: 27
+nebula: 26
+young: 22
+glow: 22
+blue: 20
+clouds: 20
+milky way: 19
+spiral galaxy: 18
+wide field camera 3: 18
+disc: 17
+sun: 17
+dark: 16
+planetary: 16
+hydrogen: 15
+star formation: 15
+infrared: 12
+advanced camera for surveys: 12
+glowing: 12
+elliptical: 11
+radiation: 11
+wfc3: 11
+gravitation: 10
+star forming: 10
+star cluster: 9
+arc: 9
+messier: 9
+ultraviolet: 9
+globular cluster: 8
+globular clusters: 8
+supernova: 8
+eso: 8
+merge: 7
+green: 7
+planetary nebula: 7
+starburst: 7
+moon: 6
+barred spiral galaxy: 6
+dust lane: 6
+acs: 6
+black hole: 6
+turbulent: 6
+atmosphere: 6
+dust lanes: 5
+star clusters: 5
+carina: 5
+comet: 5
+aging: 5
+reddish: 5
+hunter: 5
+galaxy cluster: 5
+pillar: 5
+large magellanic cloud: 5
+saturn: 5
+orange: 4
+halo: 4
+nebulae: 4
+orion: 4
+pink: 4
+collection: 4
+starburst galaxy: 4
+solar system: 4
+stellar nursery: 4
+coma berenices: 3
+colliding: 3
+supermassive: 3
+supermassive black hole: 3
+satellite: 3
+planetary nebulae: 3
+tilt: 3
+white dwarf: 3
+tint: 3
+hubble servicing mission: 3
+stellar wind: 3
+great red spot: 3
+tendril: 3
+tendrils: 3
+carina nebula: 3
+deep field: 2
+pinwheel: 2
+draco: 2
+puppis: 2
+lepus: 2
+small magellanic cloud: 2
+satellite galaxy: 2
+filaments: 2
+filament: 2
+web: 2
+lacy: 2
+pegasus: 2
+twins: 2
+tinted: 2
+arcs: 2
+grand: 2
+hidden treasures: 2
+stellar winds: 2
+voyager: 2
+supernovae: 2
+dorado: 2
+white dwarfs: 2
+variable star: 2
+variable: 2
+ngc 6302: 2
+butterfly nebula: 2
+butterfly: 2
+virgo: 2
+cetus: 2
+eagle nebula: 2
+tarantula nebula: 2
+crab nebula: 2
+crab: 2
+thread: 2
+ngc4826: 1
+black eye: 1
+evil eye: 1
+ngc 1672: 1
+ugc 10214: 1
+3b: 1
+hubble deep field: 1
+tadpole: 1
+ngc 2525: 1
+ngc2525: 1
+messier 79: 1
+messier'€™s: 1
+ngc 346: 1
+smc: 1
+ngc 7049: 1
+indus: 1
+pn m2: 1
+m2: 1
+twin jet nebula: 1
+bipolar planetary nebula: 1
+outburst: 1
+binary: 1
+binary system: 1
+ngc 3717: 1
+hydra: 1
+sea serpent: 1
+ngc 7331: 1
+winged horse: 1
+whirlpool: 1
+ngc 2841: 1
+horsehead nebula: 1
+horsehead: 1
+barnard: 1
+m81: 1
+1993j: 1
+messier 94: 1
+ngc 2371: 1
+galaxy clusters: 1
+brightest cluster galaxy: 1
+bcg: 1
+sloan digital sky survey: 1
+sdss j1156+1911: 1
+greenish: 1
+ngc 7320: 1
+7318a: 1
+7318b: 1
+natal cloud: 1
+n90: 1
+messier 106: 1
+ic 418: 1
+ngc 3201: 1
+ngc 7635: 1
+ic 10: 1
+ugca 193: 1
+sextans: 1
+sextant: 1
+abell 1703: 1
+westerlund 2: 1
+022058s: 1
+einstein rings: 1
+einstein ring: 1
+molten ring: 1
+new horizons: 1
+spacecraft: 1
+turbulence: 1
+m74: 1
+eso 510: 1
+eso 381: 1
+ngc 2022: 1
+ngc 4038: 1
+antennae: 1
+antennae galaxies: 1
+ngc 1792: 1
+ngc 5866: 1
+ngc 2264: 1
+m16: 1
+cone nebula: 1
+europa: 1
+ngc 1805: 1
+n11: 1
+messier 57: 1
+variable stars: 1
+rs puppis: 1
+cepheid: 1
+cepheid variable: 1
+north pole: 1
+auroras: 1
+hh 901: 1
+ngc 1052: 1
+ngc 6960: 1
+ngc 4621: 1
+messier 59: 1
+virgin: 1
+ngc 3256: 1
+outer planet atmospheres legacy: 1
+opal: 1
+messier 77: 1
+pinwheeling: 1
+pillars of creation: 1
+ngc 4676: 1
+mice: 1
+v838 monocerotis: 1
+v838 mon: 1
+ngc 1015: 1
+black holes: 1
+whale: 1
+catherine wheel: 1
+ngc 1512: 1
+ngc 3603: 1
+ngc 3372: 1
+keyhole nebula: 1
+ngc 1999: 1
+alpha centauri: 1
+ag carinae: 1
+remnants: 1
+ngc 2008: 1
+pictor: 1
+spiralling: 1
+collections: 1
+ngc 6217: 1
+ursa major: 1
+ngc 2174: 1
+monkey head nebula: 1
+1980s: 1
+uranus: 1
+ionosphere: 1
+space telescope imaging spectrograph: 1
+stis: 1
+canis major: 1
+messier 100: 1
+messier 104: 1
+m104: 1
+sombrero galaxy: 1
+sombrero: 1
+photogenic: 1
+ngc 1316: 1
+ngc 1300: 1
+turquoise: 1
+archival pure parallel project: 1
+appp: 1
+afgl 5180: 1
+ngc 1866: 1
+ngc 1275: 1
+threads: 1
+elliptical central cluster galaxy: 1
+m61: 1
+european southern observatory: 1
+ngc 5128: 1
+centaurus a: 1
+bug nebula: 1
+hourglass: 1
+ngc 7714: 1
+lagoon nebula: 1
+messier 82: 1
+m82: 1