Pytorch-UNet and unet

Pytorch-UNet
51
Established
unet
49
Emerging
Maintenance 0/25
Adoption 10/25
Maturity 16/25
Community 25/25
Maintenance 0/25
Adoption 10/25
Maturity 16/25
Community 23/25
Stars: 11,266
Forks: 2,731
Downloads: —
Commits (30d): 0
Language: Python
License: GPL-3.0
Stars: 268
Forks: 86
Downloads: —
Commits (30d): 0
Language: Jupyter Notebook
License: GPL-3.0
Stale 6m No Package No Dependents
Stale 6m No Package No Dependents

About Pytorch-UNet

milesial/Pytorch-UNet

PyTorch implementation of the U-Net for image semantic segmentation with high quality images

Supports mixed precision training (FP16) and automatic gradient scaling for memory efficiency on modern GPUs, plus multiclass segmentation tasks beyond the original Carvana dataset. Includes Weights & Biases integration for real-time training visualization and a pretrained model loadable via torch.hub, with Docker containerization for reproducible environments.

About unet

jakeret/unet

Generic U-Net Tensorflow 2 implementation for semantic segmentation

Related comparisons

Pytorch-UNet and u-net Pytorch-UNet and tf_unet Pytorch-UNet and unet Pytorch-UNet and tf_unet

Scores updated daily from GitHub, PyPI, and npm data. How scores work