← all repositories

ShawnBIT/UNet-family

A curated collection of UNet-family neural network architectures implemented in PyTorch for semantic image segmentation.

2.6k stars Python Computer VisionML Frameworks
UNet-family
Velocity · 7d
+0.9
★ / day
Trend
steady
star history

This repository compiles semantic segmentation models based on the UNet architecture, primarily implemented in PyTorch. It includes various UNet variants such as U-Net, V-Net, 3D U-Net, and H-DenseUNet spanning from 2015 onward, with links to original papers and alternative framework implementations. The focus is on medical image segmentation applications, and the project serves as a reference library for deep learning practitioners working on biomedical imaging tasks.

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