← all repositories

LeeJunHyun/Image_Segmentation

PyTorch implementations of U-Net, R2U-Net, Attention U-Net, and Attention R2U-Net for biomedical image segmentation.

3.1k stars Python Computer VisionML Frameworks
Image_Segmentation
Velocity · 7d
+1.1
★ / day
Trend
steady
star history

This repository provides PyTorch implementations of four convolutional neural network architectures for image segmentation. The models include standard U-Net, recurrent residual U-Net (R2U-Net), attention U-Net, and their combined variant (Attention R2U-Net). Evaluations were conducted on the ISIC 2018 skin lesion segmentation dataset.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.