← all repositories

bubbliiiing/unet-pytorch

A PyTorch implementation of the U-Net convolutional neural network for biomedical image segmentation.

1.7k stars Python Computer VisionML Frameworks
unet-pytorch
Velocity · 7d
+0.9
★ / day
Trend
steady
star history

This repository provides a complete PyTorch implementation of the U-Net architecture for semantic image segmentation. It supports multiple backbone networks (VGG, ResNet), includes training and evaluation pipelines with mIOU metrics, and is optimized for biomedical datasets with features like adjustable learning rate schedules and optimizer choices.

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