MoleImg / Attention_UNet

Raw implementation of attention gated U-Net by Keras
145 stars 43 forks source link
attention-mechanism computer-vision

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).