mirapy-org / mirapy

MiraPy: A Python package for Deep Learning in Astronomy
https://mirapy.org/
MIT License
44 stars 10 forks source link

HTRU1 Pulsar Image Classification #16

Closed swapsha96 closed 5 years ago

swapsha96 commented 5 years ago

Recently, Anna Scaife released HTRU1 dataset which was originally used to train SPINN pulsar classifier. She also shared code for classification of pulsar and non-pulsar using CNN in Pytorch. I guess can integrate the analysis with MiraPy since the structure of dataset is very similar to MNIST dataset so we can look into other applications of Deep Learning.

The first and foremost task would be writing a DataGenerator class using keras. The neural network implementation has a scope of improvement through data augmentation. We would look into other possible ways to create the 'state of the art' technique for the same. This is a really good problem for first timers.

Anna's tweet Her Notebook