Skip to content

Latest commit

 

History

History
7 lines (4 loc) · 770 Bytes

README.md

File metadata and controls

7 lines (4 loc) · 770 Bytes

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.