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

Comparisons in this category

Pytorch-UNet and tf_unet (51 vs 44) Pytorch-UNet and u-net (51 vs 44) Pytorch-UNet and unet (51 vs 43) Pytorch-UNet and unet (51 vs 42) tf_unet and unet (44 vs 42) u-net and unet (44 vs 43)