Code structure

• train.ipynb Trains our method and the baselines and generates all the figures illustrated in the paper.
• prepare_data.py Preprocesses the images and saves the result in the "galaxy_data.pkl".
• /results Contains the trained model of our method and the baselines.