Closed andrevitorelli closed 3 years ago
I added a BuilderConfig
class to the gal_gen.py
dataset file that enables to creates variants.
We can specify for instance stamp size, pixel size or the flux, parameters that are used to create galaxy images with datasets/galaxies.py
.
Specifying the name and parameters of your variant such as:
GalGenConfig(name='variant1', stamp_size=50, pixel_scale=.2, flux=1.)
you can call it with:
tfds.load('GalGen/variant1')
\o/ this is great :-D Thanks @b-remy
Many thanks @andrevitorelli and @b-remy ! Let's use this dataset as our baseline for now, we can add additional features as they are needed for the next steps of the project, for instance PSF and/or Fourier image of the galaxies.
I'm going to go ahead and merge this :-)
I think it's looking good :-) but could you update your branch to
main
? so that I can see if it works when the package is installed?