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
Rasa breast cancer radiology AI chatbot to help doctor segment lesions using Unity, Keras Attention UNet, LinkNet, etc
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).
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.
Deep Learning with CNNs course project
The official repository accompanying the paper "Deep Vision-Based Framework for Coastal Flood Prediction Under Climate Change Impacts and Shoreline Adaptations".
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."