Galaxy2Galaxy, or G2G for short, is a library of models, datasets, and utilities to build generative models for astronomical images, based on the Tensor2Tensor library. Similarly to T2T, the goal of this project is to accelerate research in machine learning models applied to astronomical image processing problems.
G2G can easily be installed using pip:
$ pip install galaxy2galaxy[galsim]
Should the install fail because of the GalSim dependency, check the GalSim install guide to fix it.
To generate the COSMOS 25.2 sample at native pixel scale and stamp size:
$ g2g-datagen --problem=img2img_cosmos --data_dir=data/img2img_cosmosThis uses GalSim to draw postage stamps and save them in TFRecord format which can then be used for training.
To train an autoencoder with this data:
$ g2g-trainer --data_dir=data/img2img_cosmos --output_dir=training/cosmos_ae --problem=img2img_cosmos --model=continuous_autoencoder_basic --train_steps=2000 --eval_steps=100 --hparams_set=continuous_autoencoder_basic