Forging new worlds (GANaxies)

High-resolution synthetic galaxies with chained generative adversarial networks

This repository contains supplementary code for the paper Forging new worlds: high-resolution synthetic galaxies with chained generative adversarial networks published in MNRAS, a preprint version of which can be found on arXiv.

For an easily accessible application, this repository provides the trained models for generating galaxies using our chained GAN model. Please note that you will need a functioning PyTorch installation to make use of the code.

File description

• generate_galaxies.py: The user-facing Python file to create synthetic galaxy samples

• models.py: The DCGAN and StackGAN models used in the corresponding research and paper

• dcgan_G.pth: The pre-trained DCGAN generator used to create synthetic 64x64 pixel galaxy images

• stackgan_G.pth: The pre-trained StackGan generator used to upscale the DCGAN generator output

Quickstart guide

Generating galaxies with the provided pretrained models works in both Python 2 and Python 3. In order to create one synthetic galaxy image, you can simply use the following command:

python generate_galaxies.py

If you want to create a specific number of synthetic samples, modify the batchSize parameter with that number:

python generate_galaxies.py --batchSize 42

Below, you can see a few examples of galaxy images created with the pretrained models.