/Attention_UNet

Raw implementation of attention gated U-Net by Keras

Primary LanguagePython

Attention_UNet

Raw implementation of attention gated U-Net using Keras. Including:

AttentionResUNet:

U-Net model with residual block, using the spatial-level attention gate.

AttentionSEResUNet:

U-Net model with residual block, using both the spatial-level and channel-level attention gate (similar to SENet).