This repository contains the implementation of 2D UNet architecture for fetal brain segmentation
(https://arxiv.org/pdf/1505.04597.pdf)
First figure shows raw MR image, second hand annotated groundtruth image and last shows weight map for spatial weighted cross entropy loss
git clone https://github.com/koriavinash1/Fetal-Brain-Segmentation.git
cd Fetal-Brain-Segmentation
pip install -r requirements.txt
Run Generate_Procesed_Data notebook for generating pre-processed data
./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
python runner.py
for training and python predictor.py
for testing the model
If any comments or issues, pull requests/issues are Welcomed....
Thankyou
- Avinash Kori (avinashgkori@smail.iitm.ac.in)