Tensorflow 2 code for several U-Net variants to perform direct comparisons including base, attention, dense, ++, squeeze-excite, inception, residual, recurrent-residual
The code here is inspired by and builds off a number of previous implementations:
- zhixuhao, Implementation of deep learning framework -- Unet, using Keras, https://github.com/zhixuhao/unet (2019).
- clguo, Fully Dense UNet implementation in medical image segmentation, https://github.com/clguo/Dense_Unet_Keras (2019).
- Xu, X. Attention U-Net. https://www.kaggle.com/xxc025/attention-u-net (2019).
- titu1994, Implementation of Squeeze and Excitation Networks in Keras, https://github.com/titu1994/keras-squeeze-excite-network (2020).
- S. Taghipour, UNet_ResNet34, https://www.kaggle.com/saeedtqp/unet-resnet34, (2019).
- lixiaolei1982, Keras Implementation of U-Net, R2U-Net, Attention U-Net, Attention R2U-Net, https://github.com/lixiaolei1982/Keras-Implementation-of-U-Net-R2U-Net-Attention-U-Net-Attention-R2U-Net.- (2019).
- longpollehn, A Keras implementation of UNet++, https://github.com/longmakesstuff/UnetPlusPlus, (2021).
- danielenricocahall, Keras-UNet, https://github.com/danielenricocahall/Keras-UNet, (2019).