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
Mar 11, 2025 - 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.
✨ Dive into image denoising magic! This project uses Attention U-Net and PatchGAN to tackle noise types like low Gaussian and salt-and-pepper noise. Perfect for computer vision, deep learning, and generative modeling enthusiasts. Restore clarity to noisy images with cutting-edge AI! 🚀🎨
Brain tumor segmentation
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.
Tackling the challenges of off-road environment navigation through Attention Guided Off-Road Semantic Segmentation.
Welcome, Awesome-segmentation-in-medical project. This repo is robust, reproducible, and fair benchmarking framework for breast ultrasound (BUS) image segmentation. Features 10+ models (UNet, UNet++, TransUnet, SwinUnet etc.), 5+ datasets, and strict data separation to prevent test set leakage.
A Satellite Semantic Segmentation Project using Unet and Attention Unet with Pytorch,
Rasa breast cancer radiology AI chatbot to help doctor segment lesions using Unity, Keras Attention UNet, LinkNet, etc
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
Deep neural networks in structural health monitoring
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.
Projects based on self learning purpose.
The official repository accompanying the paper "Deep Vision-Based Framework for Coastal Flood Prediction Under Climate Change Impacts and Shoreline Adaptations".
PyTorch implementation of UNet, UNet++, Attention UNet
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."