attention-unet 开源项目推荐（81 个）

yingkaisha /keras-unet-collection

TensorFlow、Keras实现U-net、V-net、U-net++、UET 3+、Attention U-net、R2 U-net、ResUnet-a、U#2-Net、TransUET和Swin-UET，并带有可选的ImageNet训练骨干。

arkanivasarkar /Retinal-Vessel-Segmentation-using-variants-of-UNET

Retinal vessel segmentation using U-NET, Res-UNET, Attention U-NET, and Residual Attention U-NET (RA-UNET)

DIAGNijmegen /prostateMR_3D-CAD-csPCa

层级概率3D U-Net，具有注意力机制（—Attention U-Net、SEResNet）和带有深度监督的嵌套解码器结构（—UNet++）。内置于 TensorFlow 2.5。配置为多通道3D血压MRI扫描中临床显著的体素级别前列腺癌检测。

afiosman /attention-aware-3D-UNet-for-RT-dose-prediction

Evaluate the feasibility of attention-gated 3D U-Net for knowledge-based planning dose prediction of head-and-neck radiotherapy.

afiosman /deep-learning-KBP-methods-for-rt-dose-prediction

Compare various deep learning-based knowledge-based planning methods for 3D dose distribution prediction of the head and neck plans.

rizalmaulanaa /Robustness_of_Prob_U_Net

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

Jonghwan-dev /Awesome-Segmentation-in-Medical

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.

MahanVeisi8 /From-Chaos-to-Clarity-Denoising-Images-with-UNet-and-GANs

✨ 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! 🚀🎨

davinmaritza /davion-ai

Clinical Neuro-Oncology Diagnostic Suite & Cranial MRI CADx Workstation (Tauri 2.0 + React 18 + PyTorch, N=2,800 Holdout Validation, 87.68% Multi-Class Acc)

mahsaama /BrainTumorSegmentation

Brain tumor segmentation

LiganiumInc /Satellite-Semantic-Segmentation

A Satellite Semantic Segmentation Project using Unet and Attention Unet with Pytorch,

JamesYang568 /Attention-U2Net

a dual Attention U2Net and lightweighting model

OSSome01 /semseg

Tackling the challenges of off-road environment navigation through Attention Guided Off-Road Semantic Segmentation.

Dynamo13 /Attention_UWNet

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

anirudhnbharadwaj /BraTS-2021

Brain Tumor Segmentation in Multi-Modal MRI Using Deep Learning

GivralNguyen /BKAI-IGH-Neopolyp-Segmentation

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.

ImtiazHossain-Eshan /Brain-Tumor-Segmentation-Classification

Deep learning solution for brain tumor segmentation & classification using U-Net, Attention U-Net, and advanced CNNs on the BRISC 2025 dataset. PyTorch implementation.

Arnukk /CASPIAN

The official repository accompanying the paper "Deep Vision-Based Framework for Coastal Flood Prediction Under Climate Change Impacts and Shoreline Adaptations".

AI-Medical-Robotics /Breast-Cancer-Segmentation

Rasa 乳腺癌放射学 AI 聊天机器人，使用 Unity、Keras Attention UNet、LinkNet 等帮助医生分割病灶

msikorski93 /Breast-Cancer-Ultrasound-Image-Segmentation-With-Attention-U-Net

An image segmentation problem completed with Attention U-Net network on breast ultrasound images.