This repository contains the CNN model as well as a subset of multispectral SDSS images for testing.
Check out J. Pasquet et al. 2018 for a detailed description of the model and its performance.
The test.py
Python code runs the CNN in inference mode on 100 example images (stored in data/data_example.npz
) using a set of pretrained weights. The code computes a photometric redshift estimate and the associated PDF over 180 redshift bins for each of the provided galaxy images.
Running the code It is as simple as
python test.py
The code has been tested with Python 2.7 and Python 3.6. The CNN model is built on top of the TensorFlow framework and should be compatible with TF versions ≥ 1.4.1. The Matplotlib and NumPy packages are also required for running the code
As the CNN weights exceed GitHub's regular 100MB filesize limit, you must first download the zipped pretrained_model/
folder from here or here and unzip it in the photoz
directory.