Segmentation of coronal holes and active regions using SDO/AIA images.
- data
- ar_custom.zip - Contains files of manually annotated active regions.
- ch_custom.zip - Contains files of manually annotated coronal holes.
- sample_data.zip - Contains 100 193Å images + CH binary masks and 100 171Å + AR binary masks.
- notebooks
- ar_segmentation.ipynb - Segmentation of active regions.
- ch_segmentation.ipynb - Segmentation of coronal holes.
- data_custom.ipynb - Downloading images and creating binary masks based on coordinates in
ar_custom.zipandch_custom.zip.
- src
- metrics.py - Dice and IoU metrics.
- model_scss_net.py - Segmentation model called by function
scss_net. - prep_utils.py - Data preparation utilities.
- utils.py - Utilities for plotting.
List of all used libraries can be found in requirements.txt file.
- Albumentations - library used for data augmentations.
- ImageDataAugmentor - library that extends base Keras ImageDataGenerator and allows use of custom augmentations
Data used for training and trained models can be found on this link.
The training data were images of the sun (SDO AIA) and a corresponding binary mask of a specific region (a coronal hole or active region).
You can download images with HEK API or with SunPy module. Additional data sources can be found below:
SCSS model is based on U-Net architecture used for segmentation of biomedical images. Summary of model can be found in modelsummary.txt file. Image below shows how SCSS model architecture.
Clicking on image opens video uploaded on YouTube.
