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.
11,266 stars. No commits in the last 6 months.
Stars
11,266
Forks
2,731
Language
Python
License
GPL-3.0
Category
Last pushed
Aug 11, 2024
Commits (30d)
0
Get this data via API
curl "https://pt-edge.onrender.com/api/v1/quality/ml-frameworks/milesial/Pytorch-UNet"
Open to everyone — 100 requests/day, no key needed. Get a free key for 1,000/day.
Related frameworks
BiaPyX/BiaPy
Open source Python library for building bioimage analysis pipelines
Abe404/root_painter
RootPainter: Deep Learning Segmentation of Biological Images with Corrective Annotation
stardist/stardist-napari
StarDist plugin for napari
Dootmaan/MT-UNet
Official Code for *Mixed Transformer UNet for Medical Image Segmentation*
chanzuckerberg/napari-hub
Discover, install, and share napari plugins