Skip to content

pietz/unet-keras

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

25 Commits
 
 
 
 

Repository files navigation

U-Net for Keras

This is an implementation of the U-Net model for Keras. When using the default parameters this will be the same as the original architecture, except that this code will use padded convolutions. Also, the original paper does not state the amount of Dropout used. Since 0.5 is a common choice in the industry, it was used as the default value. If you are aware of other differences, please contact me.

U-Net: Convolutional Networks for Image Segmentation (https://arxiv.org/abs/1505.04597)

Other than that, I added a few bells and whistles. This implementation supports transposed convolutions (deconvolutions) for upsampling, strided convolutions for downsampling, batchnormalization and residual connections. If you have further ideas, let me know.

About

U-Net model for Keras

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages