/Keras-UNet

Implementation of U-Net in Keras

Primary LanguagePython

Keras-UNet

Implementation of basic U-Net in Keras, along with an implementation of a U-Net with Inception Modules and U-Net with Dilated Inception Modules.

Inception Modules

Regular Inception Module

Architecture derived from https://arxiv.org/abs/1512.00567.

alt text

Dilated Inception Module

Architecture derived from https://arxiv.org/abs/1707.07128.

alt text

Model Diagrams

U-Net

Architecture and image derived from https://arxiv.org/abs/1505.04597. alt text

U-Net w/ Inception Modules

Architecture and image derived from https://www.frontiersin.org/articles/10.3389/fncom.2019.00044/full. alt text

U-Net w/ Dilated Inception Modules

alt text

TODO

  • Update Figures to higher quality/resolution
  • Add demo scripts