Skip to content

Implementation of U_Net architecture for medical image segmentation purpose.

Notifications You must be signed in to change notification settings

vaibhavshukla182/U_Net-for-medical-image-segmentation

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

8 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

U_Net for medical image segmentation

In this repo I have implemented the U_Net architecture as proposed in this paper - U-Net: Convolutional Networks for Biomedical Image Segmentation . The architecture in the paper is as follows: I have not implemented the architecture as it is but made some minor changes like adding dropouts etc. to improve my accuracy. The final architecture I used is as follows:

arch.py - file contains the code for the architecture

data.py - file contains all the neccesary code for data preprocessing, one can make changes in this file according to their own dataset

To train-

python main.py

Requirement:

Tensorflow>=1.13.1

Dataset used here:

2018 Data Science Bowl | Kaggle

I have trained this architecture for cell nucleus detection from images.Masks of the cell nuclei are provided with the dataset to train our model for segmentation purpose.Training image example-

Mask of the above image-


Following are the test results:


Releases

No releases published

Packages

No packages published

Languages