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.
MIST: A simple, scalable, and end-to-end framework for 3D medical imaging segmentation.
Compare various deep learning-based knowledge-based planning methods for 3D dose distribution prediction of the head and neck plans.
Brain tumor segmentation
Tackling the challenges of off-road environment navigation through Attention Guided Off-Road Semantic Segmentation.
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.
Task: Neural Style Transfer. The implemented solution uses a CycleGan architecture.
Cerebral Tumor Analysis and Segmentation Web Application
Official implementation utilised on the paper: Disagreement attention: Let us agree to disagree on computed tomography segmentation
Project based on Neural Style Transfer, implemented using a CycleGan architecture
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."