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

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

Deep neural networks in structural health monitoring

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

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.

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

Cerebral Tumor Analysis and Segmentation Web Application

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

Projects based on self learning purpose.

PyTorch implementation of UNet, UNet++, Attention UNet