Skip to content

TanyaChutani/UNet-Keras

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

31 Commits
 
 
 
 
 
 
 
 

Repository files navigation

Unet

Data-science-bowl-2018-Unet

Data

The original dataset is data science bowl 2k18 - dataset (automating nucleus detection)
and I've downloaded it, done the pre-processing and resizing all of the images to 128,128.
Dataset link - Dataset

Model


  • Unet, deep neural network is implemented with Keras functional API.
  • Output from the network is a 128*128 which represents mask that should be learned.
  • The model was trained for 20 iterations with lr=0.001.
  • Metric for the training is basically just a mean IOU.

Results

You can download the pretrained weights from here
Weights

Input Image.

Mask.