mouni13r/Foetal-Head-Segmentation

DATASET USED

Siim dataset

https://www.kaggle.com/anisayari/siimacrpneumothoraxsegmentationzip-dataset

HC18 dataset

https://zenodo.org/record/1327317#.YIUF34MzZbI

RUN CODE

UNET: Run unet_pytorch.ipynb

UNET with attention: Run unet_attn_pytorch.ipynb

DUNET: Run du-net.ipynb

SIIM dataset: Run python3 main.py

REFERENCES

• https://arxiv.org/pdf/2004.07071.pdf
• https://arxiv.org/pdf/2009.02805.pdf