Semantic Segmentation Models ML Frameworks

Implementations and frameworks for semantic segmentation tasks using deep learning architectures (DeepLab, FCNs, etc.). Includes domain-specific applications like road/aerial segmentation. Does NOT include general image classification, instance segmentation as primary focus, or road-specific datasets without model code.

There are 102 semantic segmentation models frameworks tracked. 1 score above 50 (established tier). The highest-rated is meetps/pytorch-semseg at 51/100 with 3,411 stars.

Get all 102 projects as JSON

curl "https://pt-edge.onrender.com/api/v1/datasets/quality?domain=ml-frameworks&subcategory=semantic-segmentation-models&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	meetps/pytorch-semseg Semantic Segmentation Architectures Implemented in PyTorch	51	Established	3,411	Python
2	rankseg/rankseg Boost segmentation model mIoU/Dice instantly WITHOUT retraining. A...	46	Emerging	52	Jupyter Notebook
3	MLSTRUCT/MLStructFP Multi-unit floor plan dataset for architectural analysis and recognition	45	Emerging	59	Jupyter Notebook
4	fregu856/deeplabv3 PyTorch implementation of DeepLabV3, trained on the Cityscapes dataset.	44	Emerging	816	Python
5	wkentaro/pytorch-fcn PyTorch Implementation of Fully Convolutional Networks. (Training code to...	44	Emerging	1,782	Python
6	DrSleep/light-weight-refinenet Light-Weight RefineNet for Real-Time Semantic Segmentation	44	Emerging	741	Python
7	MarvinTeichmann/MultiNet Real-time Joint Semantic Reasoning for Autonomous Driving	44	Emerging	556	Python
8	luyanger1799/Amazing-Semantic-Segmentation Amazing Semantic Segmentation on Tensorflow && Keras (include FCN, UNet,...	44	Emerging	481	Python
9	yassouali/pytorch-segmentation :art: Semantic segmentation models, datasets and losses implemented in PyTorch.	44	Emerging	1,814	Jupyter Notebook
10	sthalles/deeplab_v3 Tensorflow Implementation of the Semantic Segmentation DeepLab_V3 CNN	44	Emerging	805	Jupyter Notebook
11	zijundeng/pytorch-semantic-segmentation PyTorch for Semantic Segmentation	44	Emerging	1,740	Python
12	sithu31296/semantic-segmentation SOTA Semantic Segmentation Models in PyTorch	43	Emerging	939	Python
13	akirasosa/mobile-semantic-segmentation Real-Time Semantic Segmentation in Mobile device	43	Emerging	722	Python
14	isht7/pytorch-deeplab-resnet DeepLab resnet v2 model in pytorch	43	Emerging	603	Python
15	Tramac/Fast-SCNN-pytorch A PyTorch Implementation of Fast-SCNN: Fast Semantic Segmentation Network	43	Emerging	429	Python
16	ternaus/TernausNetV2 TernausNetV2: Fully Convolutional Network for Instance Segmentation	43	Emerging	546	Jupyter Notebook
17	TuSimple/TuSimple-DUC Understanding Convolution for Semantic Segmentation	42	Emerging	611	Python
18	bfortuner/pytorch_tiramisu FC-DenseNet in PyTorch for Semantic Segmentation	42	Emerging	305	Jupyter Notebook
19	iArunava/ENet-Real-Time-Semantic-Segmentation ENet - A Neural Net Architecture for real time Semantic Segmentation	42	Emerging	285	Jupyter Notebook
20	MLSTRUCT/MLSTRUCT-FP Multi-unit floor plan dataset for architectural analysis and recognition (AI)	42	Emerging	59	Jupyter Notebook
21	sacmehta/ESPNetv2 A light-weight, power efficient, and general purpose convolutional neural network	40	Emerging	454	Python
22	HasnainRaz/FC-DenseNet-TensorFlow Fully Convolutional DenseNet (A.K.A 100 layer tiramisu) for semantic...	40	Emerging	124	Python
23	arplaboratory/STHN [RA-L 2024] Official repository for "STHN: Deep Homography Estimation for...	38	Emerging	79	Jupyter Notebook
24	wiktorlazarski/head-segmentation 👦 Human head semantic segmentation	38	Emerging	89	Jupyter Notebook
25	whchien/deep-floor-plan-recognition The project uses a computer vision model to extract structured features from...	37	Emerging	73	Python
26	GeorgeSeif/Semantic-Segmentation-Suite Semantic Segmentation Suite in TensorFlow. Implement, train, and test new...	36	Emerging	2,526	Python
27	guanfuchen/semseg 常用的语义分割架构结构综述以及代码复现	36	Emerging	789	Python
28	thuyngch/Human-Segmentation-PyTorch Human segmentation models, training/inference code, and trained weights,...	35	Emerging	574	Jupyter Notebook
29	qqlu/Entity EntitySeg Toolbox: Towards Open-World and High-Quality Image Segmentation	34	Emerging	1,041	Jupyter Notebook
30	Angzz/awesome-panoptic-segmentation Panoptic Segmentation Resources List	34	Emerging	556	—
31	aj1365/PolSARFormer This code is for the paper "Local Window Attention Transformer for...	34	Emerging	57	Jupyter Notebook
32	IbrahimSobh/Segmentation In this tutorial, you will perform inference across 10 well-known...	34	Emerging	75	Jupyter Notebook
33	edwardyehuang/CAR CAR: Class-aware Regularizations for Semantic Segmentation (ECCV-2022)	32	Emerging	30	Python
34	xmindflow/DAEFormer [MICCAI 2023] DAE-Former: Dual Attention-guided Efficient Transformer for...	32	Emerging	126	Python
35	celoyd/potato A small pansharpening model	31	Emerging	81	Python
36	NikolasEnt/Road-Semantic-Segmentation Udacity Self-Driving Car Engineer Nanodegree. Project: Road Semantic Segmentation	31	Emerging	92	Jupyter Notebook
37	conradry/max-deeplab Unofficial implementation of MaX-DeepLab for Instance Segmentation	31	Emerging	112	Jupyter Notebook
38	reecehuff/StrainNet StrainNet is a deep neural network for measuring deformation from images	31	Emerging	12	Python
39	mahdizynali/SegLight Seglight is super fast and real-time semantic segmentation (cpu only) that...	31	Emerging	12	Python
40	drkostas/3D-Semantic-Segmentation Semantic Segmentation with Transformers on 3D Medical Images	30	Emerging	67	Jupyter Notebook
41	maikpaixao/deep-floorplan-recognition State-of-the-art Floor Plan Recognition System. This innovative software...	30	Emerging	27	Python
42	UAws/CV-3315-Is-All-You-Need CV′3315 Is All You Need – Semantic Segmentation Course Competition @ The...	30	Emerging	22	Jupyter Notebook
43	vinceecws/SegNet_PyTorch PyTorch implementation of SegNet: A Deep Convolutional Encoder-Decoder...	29	Experimental	75	Jupyter Notebook
44	statmlben/rankseg RankSEG: A consistent ranking-based framework for segmentation	28	Experimental	30	Jupyter Notebook
45	Jittor/JSeg JSeg is a Semantic segmentation toolbox based on MMSegmentation and Jittor	28	Experimental	225	Python
46	ademakdogan/nails_segmentation This project was made for nails segmentation using deep learning models....	28	Experimental	43	Python
47	Followb1ind1y/Semantic-Segmentation-of-Aerial-Imagery Semantic Segmentation of Aerial Imagery Project using PyTorch	28	Experimental	9	Jupyter Notebook
48	matlab-deep-learning/pretrained-deeplabv3plus DeepLabv3+ inference and training in MATLAB for Semantic Segmentation	28	Experimental	24	MATLAB
49	Ginga-Kennis/semantic_segmentation_ros ROS package for semantic segmentation	27	Experimental	32	Python
50	wentacc/Bayesian-Segmentation-and-Uncertainty-estimation-on-CityScapes Implement a network for semantic segmentation in image data, and also...	27	Experimental	113	Jupyter Notebook
51	tobiassteidle/ML_IDCard_Segmentation_Pytorch Machine Learning Project to identify an ID Card on an image	27	Experimental	64	Python
52	BMW-InnovationLab/BMW-Semantic-Segmentation-Inference-API-GPU-CPU This is a repository for a Semantic Segmentation inference API using the...	26	Experimental	57	Python
53	KostasStefanidis/Semantic-Segmentation Semantic Segmentation for Urban Scene understanding - Cityscapes dataset	26	Experimental	11	Python
54	oskarnatan/semseg-bisaai GPU-accelerated Semantic Image Segmentation with PyTorch	26	Experimental	8	Jupyter Notebook
55	matciotola/Lambda-PNN Unsupervised Deep Learning-based Pansharpening with Jointly-Enhanced...	25	Experimental	30	Python
56	kerrgarr/SemanticSegmentationCityscapes A simple image segmentation model called ‘my_FCN’ is compared with a...	25	Experimental	12	Jupyter Notebook
57	theobdt/aerial_pc_classification Segmentation of urban aerial point clouds with Deep Learning in Pytorch.	25	Experimental	31	Python
58	Mr-TalhaIlyas/SegNext SegNext Implementation in PyTorch	24	Experimental	60	Python
59	ACSEkevin/An-Overview-of-Segformer-and-Details-Description This repository provides an overview of Segformer, architecture encoder in...	23	Experimental	19	Python
60	mukund-ks/DeepLabV3-Segmentation A DeepLab V3+ Model with choice of Encoder for Binary Segmentation....	23	Experimental	19	Python
61	tonysy/plusseg ShanghaiTech PLUS Lab Segmentation Toolbox and Benchmark	23	Experimental	22	Python
62	bhanderisavan/roadside-radar-seg Official repository of "Deep segmentation of 3+1D radar point cloud for...	23	Experimental	1	Python
63	farhad-dalirani/Satellite-Imagery-Road-Segmentation Semantic segmentation of road networks in high-resolution satellite images...	23	Experimental	6	Python
64	Sudheendra-Peddiraju/ADE20k-Lightweight-Semantic-Segmentation Exploring lightweight MobileNet-based semantic segmentation models on ADE20K...	22	Experimental	—	Jupyter Notebook
65	aleguillou1/SemanticSeg4EO A unified PyTorch framework for semantic segmentation of satellite imagery....	22	Experimental	19	Python
66	Quentin18/road-segmentation Road segmentation using CNNs	22	Experimental	7	Jupyter Notebook
67	mukund-ks/DeepLabV3Plus-PyTorch A DeepLab V3+ Model with ResNet 50 Encoder to perform Binary Segmentation...	22	Experimental	23	Python
68	balnarendrasapa/road-detection This is a course project for DSCI-6011 - Deep Learning. deals with Drivable...	21	Experimental	8	Jupyter Notebook
69	tonysy/DRN-MXNet Dense Relation Network: Learning Consistent and Context-Aware Representation...	21	Experimental	25	Python
70	VinayTeki/Semantic_Segmentation KERAS: Multimodal Deep Learning for Semantic Segmentation (RGB, NIR Streams)...	21	Experimental	11	Python
71	robail/Pytorch-Convolution-neural-network-for-semantic-segmentation Pytorch Convolution neural network for semantic segmentation	21	Experimental	9	Python
72	potato-farm/alchera-ai-challenge Body Part Semantic Segmentation	20	Experimental	8	Jupyter Notebook
73	RElbers/region-mutual-information-pytorch PyTorch implementation of the Region Mutual Information Loss for Semantic...	20	Experimental	26	Python
74	filippogiruzzi/semantic_segmentation Semantic Segmentation project for Autonomous Driving based on a TensorFlow...	20	Experimental	11	Python
75	markus-k/bisenetv2-tf2 BiSeNet V2 TensorFlow 2 Implementation capable of running on Edge TPUs	20	Experimental	16	Jupyter Notebook
76	Dalageo/road-segmentation-idd Distinguishing Urban Roads from Non-Road Regions in the Indian Driving...	19	Experimental	8	Jupyter Notebook
77	yc-cui/PreMix [ISPRS 2025] Pansharpening via predictive filtering with element-wise feature mixing	19	Experimental	7	Python
78	Sumanth181099/DeepMAO This is the official implementation of "DeepMAO: Deep Multi-scale Aware...	18	Experimental	11	Python
79	iurada/neutron-segmentation Official repository of our work "Transient Fault Tolerant Semantic...	18	Experimental	7	Python
80	DanielMPMatCom/Past2Polygon-ML-Project A Machine Learning project for georeferencing and digitizing maps of Havana...	17	Experimental	4	C
81	moatifbutt/r2s100k we introduce R2S100K---a large-scale dataset and benchmark for training and...	17	Experimental	15	Jupyter Notebook
82	AndrewColligan/FeatureNet-Tensorflow-2 This project is a re-implementation of FeatureNet using Tensorflow 2. With...	16	Experimental	4	—
83	project4earth/TerraSegNet TerraSegNet: Bilateral Axial Attention Network for Remote Sensing Image...	15	Experimental	—	Python
84	r-wenger/MultiFranceFences Code for MultiFranceFences dataset and benchmark models.	15	Experimental	5	Python
85	Lecheng-Wang/Awesome-Segmention-Model A unified semantic segmentation model manager integrating 14...	15	Experimental	8	Python
86	umitkacar/awesome-segmentation-all-types 📊 Comprehensive collection of segmentation techniques - semantic, instance,...	15	Experimental	—	—
87	1062270094zsl-cloud/transformer-few-shot-segmentation Transformer Few-Shot Semantic Segmentation - 评估不同监督级别的 Transformer 骨干对...	15	Experimental	—	—
88	tajwarchy/semantic-segmentation Semantic segmentation on PASCAL VOC 2012 using DeepLabV3+ ResNet50. Achieves...	14	Experimental	—	Python
89	cool-machine/transformer-based-image-segmentation AI-Powered Image Segmentation System - Production-ready computer vision...	14	Experimental	—	Jupyter Notebook
90	NAOC-LAMOST/fcn4flare official implementation of the paper: "FCN4Flare: Fully Convolution Neural...	13	Experimental	7	Python
91	amramer/Semantic-Segmentation-for-Autonomous-Vehicles This project focuses on semantic segmentation using the BDD100K dataset, a...	13	Experimental	5	Jupyter Notebook
92	yc-cui/Pansharpening-Zoo A collection of deep learning based pansharpening models.	12	Experimental	25	—
93	KentaItakura/Semantic-segmentation-using-Pascal-VOC-with-MATLAB This example shows how to train a semantic segmentation network using deep learning.	12	Experimental	4	MATLAB
94	ApaxPhoenix/Segmentation A PyTorch framework for semantic segmentation supporting 8 architectures...	11	Experimental	—	Python
95	lmtszrl/plantar_segmentation PyTorch-based Deep Learning pipeline for 8-region plantar pressure segmentation.	11	Experimental	—	Python
96	theveryhim/DeepSVG-Segmentation Implementing paper Deep Gaussian mixture model for unsupervised image segmentation	11	Experimental	—	Jupyter Notebook
97	Dhanvika27/road-scene-segmentation Semantic segmentation of road scenes using U-Net with EfficientNetB0 on the...	11	Experimental	—	Jupyter Notebook
98	francescopiocirillo/semantic-segmentation-deeplabv3-pytorch 🤖 Comprehensive study & optimization of semantic segmentation models for...	11	Experimental	—	Jupyter Notebook
99	AndrewDarnall/MSLesSeg-4-ICPR Proposed SegFormer3D and SegFormer3DMoE-based architectures for automated...	11	Experimental	—	Shell
100	shauryagh/offroad-navigation-intelligence-system Off-road navigation intelligence system using semantic segmentation to help...	11	Experimental	—	—
101	GOKULPRASANTH-M/ViMbaFormer-A-Hybrid-Vision-Mamba-Unet-Architecture-for-UAV-Semantic-Segmentation ViMbaFormer is a hybrid semantic segmentation framework designed for...	11	Experimental	—	—
102	Simon-Bertrand/SAROPTReg-Insights Associated repository to our paper on learning-based SAR-OPT registration insights	11	Experimental	—	Jupyter Notebook

Comparisons in this category

pytorch-semseg and pytorch-segmentation (51 vs 44) pytorch-semseg and pytorch-semantic-segmentation (51 vs 44) pytorch-semseg and Fast-SCNN-pytorch (51 vs 43) pytorch-semseg and pytorch_tiramisu (51 vs 42) pytorch-semantic-segmentation and pytorch-segmentation (44 vs 44) deeplabv3 and deeplab_v3 (44 vs 44)