The Tensorflow, Keras implementation of U-net, V-net, U-net++, UNET 3+, Attention U-net, R2U-net, ResUnet-a, U^2-Net, TransUNET, and Swin-UNET with optional ImageNet-trained backbones.
-
Updated
Feb 22, 2023 - Python
The Tensorflow, Keras implementation of U-net, V-net, U-net++, UNET 3+, Attention U-net, R2U-net, ResUnet-a, U^2-Net, TransUNET, and Swin-UNET with optional ImageNet-trained backbones.
Retinal vessel segmentation using U-NET, Res-UNET, Attention U-NET, and Residual Attention U-NET (RA-UNET)
Hierarchical probabilistic 3D U-Net, with attention mechanisms (—𝘈𝘵𝘵𝘦𝘯𝘵𝘪𝘰𝘯 𝘜-𝘕𝘦𝘵, 𝘚𝘌𝘙𝘦𝘴𝘕𝘦𝘵) and a nested decoder structure with deep supervision (—𝘜𝘕𝘦𝘵++). Built in TensorFlow 2.5. Configured for voxel-level clinically significant prostate cancer detection in multi-channel 3D bpMRI scans.
Evaluate the feasibility of attention-gated 3D U-Net for knowledge-based planning dose prediction of head-and-neck radiotherapy.
This is provisional codes for a conference paper titled "Robustness of Probabilistic U-Net for Automated Segmentation of White Matter Hyperintensity in Different Datasets of Brain MRI".
Compare various deep learning-based knowledge-based planning methods for 3D dose distribution prediction of the head and neck plans.
a dual Attention U2Net and lightweighting model
"Attention UW-Net: A fully connected model for automatic segmentation and annotation of chest X-ray" by Debojyoti Pal, Pailla Balakrishna Reddy, and Sudipta Roy.
Brain tumor segmentation
Deep neural networks in structural health monitoring
Tackling the challenges of off-road environment navigation through Attention Guided Off-Road Semantic Segmentation.
Machine Learning 2 Course Project at RKMVERI, 2021. Published at The Imaging Science Journal (2023), Paper: https://www.tandfonline.com/doi/full/10.1080/13682199.2023.2174657
3D U-net, Attention U-net, Res U-net, Attention Res U-net, and MSRes U-net are implemented and compared for emulation of current density induced during transcranial direct current stimulation (tDCS).
Rasa breast cancer radiology AI chatbot to help doctor segment lesions using Unity, Keras Attention UNet, LinkNet, etc
Baseline model for BKAI-IGH_Neopolyp. Currently supports Unet and Attention Unet with VGG-16, MobilenetV2 and Efficientnet-B0 backbone. This repository is private therefore not a official implementation from BKAI.
Cerebral Tumor Analysis and Segmentation Web Application
An image segmentation problem completed with Attention U-Net network on breast ultrasound images.
The official repository accompanying the paper "Deep Vision-Based Framework for Coastal Flood Prediction Under Climate Change Impacts and Shoreline Adaptations".
Deep Learning with CNNs course project
Add a description, image, and links to the attention-unet topic page so that developers can more easily learn about it.
To associate your repository with the attention-unet topic, visit your repo's landing page and select "manage topics."