vairodp / astronet Goto Github PK

Deep Learning Approaches to the SKA Science Data Challenge 1
Explore the docs »

In this project we developed a series of deep learning models to detect and classify astronomical sources from radio images. In particular, we approached the problem both as on object detection and as an image segmentation task, implementing from scratch both YOLOv4 and U-Net. The first network did not achieve an optimal performance (partially due to shortage of training data), while the latter reached a an accuracy of 97,3%.

The code requires python >= 2.7 as well as the following python libraries:

• astropy
• imgaug
• matplotlib
• numpy
• pandas
• scikit-learn
• tensorflow
• tensorflow-datasets
• tqdm
• opencv-python

Install Modules:

  pip install -U pip
  pip install -r requirements.txt

Try Demos

• AstroNet_U-Net Notebook
• AstroNet_YOLOv4 Notebook

Martina Rossini - mwritescode - [email protected]

Vairo Di Pasquale - vairodp - [email protected]