Fetal-Brain-Segmentation

Introduction

This repository contains the implementation of 2D UNet architecture for fetal brain segmentation

Network Architecture

(https://arxiv.org/pdf/1505.04597.pdf)

Raw data

---

Results

Model predictions

Dice score with epochs

---

How to use?

git clone https://github.com/koriavinash1/Fetal-Brain-Segmentation.git
cd Fetal-Brain-Segmentation
pip install -r requirements.txt

---

Pre-Processing data

Run Generate_Procesed_Data notebook for generating pre-processed data

Folder structure

./src consists all source codes

config -> all initial configurations data_loader -> multithread data loader estimator -> model estimator class network -> network architecture definition runner -> main function

---

If any comments or issues, pull requests/issues are Welcomed....

Thankyou python runner.py for training or testing the model