AstroPile / FlatironMeeting2024

AstroPile meet-up at the Flatiron Institute
https://astropile.github.io/FlatironMeeting2024/
MIT License
2 stars 3 forks source link

[Baselines] Morphology CNN for Galaxy10 DECaLS #24

Open mwalmsley opened 7 months ago

mwalmsley commented 7 months ago

Morphology CNN for Galaxy10 DECaLS

A simple small CNN to classify Galaxy10 DECaLS images.

Contacts: Mike Walmsley Participants: as above

Goals and deliverable

Script to train a simple pytorch CNN on Galaxy10 DECaLS data. Possibly some final performance measurement, depending on training time.

Resources needed

Time DECaLS images Might do it on Colab to provide an easy demo example.

Related to #22 which will grab the images

mwalmsley commented 7 months ago

Building this HF dataset to support https://huggingface.co/datasets/mwalmsley/galaxy10_decals

mwalmsley commented 7 months ago

Dataset and baseline model made. Includes demo notebook on astropile and a Colab notebook here: https://colab.research.google.com/drive/1I8Wqk48gaxFyXO7xoSEIWwY7hEoYNC1H?usp=sharing