bubbliiiing/unet-pytorch

Health v2 · maintenance only

Adoption not part of the score

1725 stars · 285 forks observed · 2026-08-28

What it is AI-extracted, prompt v1, taxonomy v1, 2026-08-30, confidence not recorded

A PyTorch implementation of the U-Net semantic segmentation model with training, prediction, and mIoU evaluation scripts. It supports multiple backbones (VGG, ResNet50) and lets users train custom models on their own VOC-format or medical imaging datasets.

Use cases

• train a custom image segmentation model on my own dataset
• segment medical images with unet
• run semantic segmentation inference with pretrained weights
• evaluate segmentation models with miou
• learn how unet works in pytorch
• fine-tune unet with vgg or resnet backbone

When to choose

• you need a well-commented, educational U-Net implementation in PyTorch
• you want to train segmentation on VOC-format or medical datasets
• you need pretrained weights and simple train/predict scripts

When to avoid

• you need state-of-the-art segmentation accuracy on complex datasets like VOC
• you want a production-ready segmentation service rather than training scripts
• you need models other than U-Net or non-PyTorch frameworks

Facets

library · maturity maintenance

machine-learning deep-learning image-processing computer-vision deep-learning computer-vision image-processing machine-learning python unet semantic-segmentation pytorch image-segmentation model-training pretrained-weights

Member repositories

For agents

markdown · JSON · MCP: product_card(name="bubbliiiing/unet-pytorch")