U-Net Segmentation PyTorch ML Frameworks

PyTorch implementations and variants of U-Net architecture for image segmentation tasks. Includes custom datasets, architectural improvements, and medical imaging applications. Does NOT include general semantic segmentation frameworks, non-U-Net architectures, or segmentation without PyTorch focus.

There are 72 u-net segmentation pytorch frameworks tracked. 9 score above 50 (established tier). The highest-rated is BiaPyX/BiaPy at 66/100 with 192 stars.

Get all 72 projects as JSON

curl "https://pt-edge.onrender.com/api/v1/datasets/quality?domain=ml-frameworks&subcategory=unet-segmentation-pytorch&limit=20"

Open to everyone — 100 requests/day, no key needed. Get a free key for 1,000/day.

#	Framework	Score	Tier	Stars	Language
1	BiaPyX/BiaPy Open source Python library for building bioimage analysis pipelines	66	Established	192	Jupyter Notebook
2	Abe404/root_painter RootPainter: Deep Learning Segmentation of Biological Images with Corrective...	58	Established	75	Python
3	junlabucsd/napari-mm3 Mother machine image analysis through napari	55	Established	10	Jupyter Notebook
4	Dootmaan/MT-UNet Official Code for Mixed Transformer UNet for Medical Image Segmentation	54	Established	196	Python
5	chanzuckerberg/napari-hub Discover, install, and share napari plugins	53	Established	54	TypeScript
6	jakeret/tf_unet Generic U-Net Tensorflow implementation for image segmentation	51	Established	1,918	Python
7	ethanhe42/u-net U-Net: Convolutional Networks for Biomedical Image Segmentation	51	Established	448	Python
8	milesial/Pytorch-UNet PyTorch implementation of the U-Net for image semantic segmentation with...	51	Established	11,266	Python
9	intel/unet U-Net Biomedical Image Segmentation	50	Established	322	Jupyter Notebook
10	stardist/stardist-napari StarDist plugin for napari	49	Emerging	39	Python
11	jakeret/unet Generic U-Net Tensorflow 2 implementation for semantic segmentation	49	Emerging	268	Jupyter Notebook
12	ImprintLab/Medical-SAM2 Medical SAM 2: Segment 3D Medical Images Via Segment Anything Model 2	48	Emerging	908	Python
13	archinetai/a-unet A toolbox that provides hackable building blocks for generic 1D/2D/3D UNets,...	47	Emerging	89	Python
14	ImprintLab/Medical-SAM-Adapter A lightweight adapter bridges SAM with medical imaging [MedIA]	47	Emerging	1,282	Python
15	jeya-maria-jose/UNeXt-pytorch Official Pytorch Code base for "UNeXt: MLP-based Rapid Medical Image...	46	Emerging	553	Python
16	lmb-freiburg/Unet-Segmentation The U-Net Segmentation plugin for Fiji (ImageJ)	45	Emerging	90	Java
17	xmindflow/LHUNet LHU-Net: A Lean Hybrid U-Net for Cost-efficient, High-performance Volumetric...	45	Emerging	65	Python
18	boschresearch/unetgan Official Implementation of the paper "A U-Net Based Discriminator for...	45	Emerging	401	Python
19	CAREamics/careamics-ui napari plugin for CAREamics	44	Emerging	7	Python
20	MarcoForte/DeepInteractiveSegmentation Getting to 99% Accuracy in Interactive Segmentation and Interactive Training...	43	Emerging	117	Jupyter Notebook
21	sinAshish/Multi-Scale-Attention [JBHI] Code for our paper "Multi-scale Guided Attention for Medical Image...	42	Emerging	466	Python
22	McGregorWwww/UDTransNet This repo is the official implementation of 'Narrowing the semantic gaps in...	40	Emerging	95	Python
23	threedle/iSeg Interactive 3D Segmentation via Interactive Attention	40	Emerging	35	Jupyter Notebook
24	jlcsilva/EfficientUNetPlusPlus Decoder architecture based on the UNet++. Combining residual bottlenecks...	39	Emerging	67	Python
25	azadef/ynet Y-Net: A Spatiospectral Dual-Encoder Networkfor Medical Image Segmentation...	38	Emerging	38	Python
26	nadeemlab/ImPartial Interactive deep learning whole-cell segmentation and thresholding using...	38	Emerging	42	Jupyter Notebook
27	chosj95/MIMO-UNet MIMO-UNet - Official Pytorch Implementation	37	Emerging	443	Python
28	xmindflow/FuseNet [ISBI 2024] FuseNet: Self-Supervised Dual-Path Network for Medical Image Segmentation	37	Emerging	22	Jupyter Notebook
29	hfawaz/ijcars19 Accurate and interpretable evaluation of surgical skills from kinematic data...	37	Emerging	10	Python
30	ajithvcoder/UNet-Pytorch-Customdataset Unet pytorch on Custom dataset	37	Emerging	13	Jupyter Notebook
31	danikiyasseh/SAIS [Nature Biomedical Engineering 2023] Decoding surgical activity from videos...	37	Emerging	22	Python
32	McGregorWwww/UCTransNet Implementation of our AAAI'22 work: 'UCTransNet: Rethinking the Skip...	36	Emerging	432	Python
33	exodusadamant/MedSAM3 🩺 Enable accurate medical image segmentation using natural language prompts...	36	Emerging	1	—
34	chanzuckerberg/saber Segment Organelles from Electron Microscopy Datasets with Segment-Anything 2	36	Emerging	15	Jupyter Notebook
35	franciszekparma/U-Net-PyTorch U-Net Full Implementation from Scratch in PyTorch for Brain Tumor Segmentation	35	Emerging	2	Jupyter Notebook
36	FengheTan9/CMU-Net [ISBI 2023] Official Pytorch implementation of "CMU-Net: A Strong...	34	Emerging	89	Python
37	FeijiangHan/CTISPED UNet series network architectures (UNet, R2UNet, Attention UNet, Nested...	34	Emerging	68	Jupyter Notebook
38	France1/unet-multiclass-pytorch Multiclass semantic segmentation using U-Net architecture combined with...	34	Emerging	62	Jupyter Notebook
39	umitkacar/awesome-unet-instance-segmentation- 🔬 Curated collection of U-Net architectures and instance segmentation...	34	Emerging	3	—
40	TopoXLab/TopoInteraction This repository contains the implementation for our work "Learning...	33	Emerging	84	Python
41	Tes-bo/UNet-CFD-FrameWork-cavity This project trains a U-Net neural network to learn the mapping from...	33	Emerging	10	C++
42	karelbecerra/sam-x-ray-medical-images-hand-sex-classification Deep learning and segmentation in sex classification from left hand X-ray...	32	Emerging	2	Jupyter Notebook
43	yl-jiang/UPerNet Pytorch Implement of UPerNet	32	Emerging	31	Python
44	nspunn1993/BT-Unet BT-Unet: A self-supervised learning framework	32	Emerging	36	Jupyter Notebook
45	clementsicard/unet-swin Swin backbone for UNet network for semantic segmentation	31	Emerging	3	—
46	jus390/U-net-Iris-segmentation The best performing Iris segmentation model from the article "An End-to-End...	29	Experimental	9	—
47	Belis0811/MaskUnet This is official github repo for InReview paper "MaskAttn-UNet: A Mask...	29	Experimental	26	Jupyter Notebook
48	lafith/Mobile-UNet PyTorch implementation of Mobile-UNet.	29	Experimental	30	Python
49	LucaLumetti/UNetTransplant Repository for the paper "U-Net Transplant: The Role of Pre-training for...	28	Experimental	31	Python
50	kyegomez/SimpleUnet An simple implementation of Unet because all the implementations i've seen...	28	Experimental	8	Python
51	ExtremeViscent/SR-UNet Customized implementation of the U-Net in PyTorch for super-resolving...	27	Experimental	19	Jupyter Notebook
52	JlexZzz/pytorch-U-Net 基于卷积神经网络U-Net实现生物医学影像分割，使用pytorch框架实现	27	Experimental	24	Jupyter Notebook
53	icon-lab/MedTrim Official implementation of "Meta-Entity Driven Triplet Mining for Aligning...	27	Experimental	14	Python
54	Ankitkumar803/U-net-for-image-segmentation-from-scratch--PyTorch U-Net	25	Experimental	7	Jupyter Notebook
55	siyi-wind/MDViT [MICCAI 2023] MDViT: Multi-domain Vision Transformer for Small Medical Image...	24	Experimental	42	Python
56	matin-ghorbani/Salt-Segmentation-UNet Implement a salt segmentation using PyTorch for TGS Salt Identification Challenge	22	Experimental	16	Python
57	Keragen/lUnet U-Net cell segmentation and tracked-spot fluorescence quantification from...	21	Experimental	—	R
58	Afnnnan/unet-human-segmentation U-Net model trained from scratch for full-body human segmentation on the...	21	Experimental	—	Jupyter Notebook
59	Ste-lla02/stella STELLA is a modular pipeline for medical image segmentation and interactive...	21	Experimental	3	Python
60	jiaowoguanren0615/DA-TransUNet This is a warehouse for DA-TransUNet-pytorch-model, can be used to train...	20	Experimental	5	Python
61	xmindflow/cenet [MICCAI 2025] CENet: Context Enhancement Network for Medical Image Segmentation	20	Experimental	25	Python
62	matin-ghorbani/Carvana-Segmentation-UNet Implement a Carvana segmentation using UNet from scratch with PyTorch	20	Experimental	7	Python
63	shervinnd/Pet_Segmentation_UNet A U-Net model for pet image segmentation using the Oxford-IIIT Pet dataset...	19	Experimental	2	Jupyter Notebook
64	kurtarmamento/prostate-unet PyTorch U-Net pipeline for MRI segmentation with reproducible...	19	Experimental	—	Python
65	a-r-n-o-l-d/UNet.jl Implementation of a generic U-Net: a convolutional neural network for image...	17	Experimental	1	Julia
66	Younessboumlik/Facial-Parts-Segmentation-with-Deep-Learning Deep learning project for facial parts segmentation using UNet, SegNet, and...	13	Experimental	—	Jupyter Notebook
67	syedhamzamohiuddin/unet-paper-reimplementation Faithful implementation of the original U-Net architecture with weight maps,...	13	Experimental	—	—
68	irfanfadhullah/unet_zoo UNet Zoo: A PyTorch library for diverse UNet architectures, inspired by...	13	Experimental	7	Python
69	eduujspeng/Deform_Adaptive_UNet Deform adaptive UNet for medical image segmentaion	12	Experimental	7	HTML
70	salwaElkaddaoui/unet Implementation of the paper "U-Net: Convolutional Networks for Biomedical...	11	Experimental	—	Python
71	bereketeshete/Microscope-Image-Super-Resolution-using-Deep-Learning-UNet Structured Illumination Microscopy (SIM) Super-Resolution using Deep Learning UNet	11	Experimental	—	—
72	ArevikKH/Segmentation-of-Handwritten-Digits-using-U-Net A completed system for segmenting handwritten digits in images using binary...	10	Experimental	1	Jupyter Notebook

Comparisons in this category

tf_unet and Pytorch-UNet (51 vs 51) Pytorch-UNet and u-net (51 vs 51) Pytorch-UNet and unet (51 vs 50) Pytorch-UNet and unet (51 vs 49) tf_unet and u-net (51 vs 51) tf_unet and unet (51 vs 50) tf_unet and unet (51 vs 49) u-net and unet (51 vs 50) u-net and unet (51 vs 49) u-net and Unet-Segmentation (51 vs 45)