← all repositories

milesial/Pytorch-UNet

A PyTorch implementation of the U-Net convolutional neural network for semantic segmentation of high-definition images.

11.5k stars Python Computer VisionML Frameworks
Pytorch-UNet
Velocity · 7d
+3.6
★ / day
Trend
steady
star history

This repository provides a customizable implementation of the U-Net architecture in PyTorch, originally designed for biomedical image segmentation and adapted here for Kaggle’s Carvana Image Masking Challenge. It includes training pipelines with mixed precision support, integration with TensorBoard and Weights & Biases for experiment tracking, Docker deployment options, and pretrained model weights.

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