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 94 semantic segmentation models frameworks tracked. 15 score above 50 (established tier). The highest-rated is MLSTRUCT/MLStructFP at 52/100 with 59 stars.

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

Comparisons in this category

pytorch-semseg and pytorch-segmentation (51 vs 51) pytorch-semantic-segmentation and pytorch-semseg (51 vs 51) pytorch-semseg and Fast-SCNN-pytorch (51 vs 50) pytorch-semseg and pytorch_tiramisu (51 vs 49) pytorch-semantic-segmentation and pytorch-segmentation (51 vs 51) pytorch-semantic-segmentation and pytorch-fcn (51 vs 51) pytorch-segmentation and semantic-segmentation (51 vs 50) pytorch-semantic-segmentation and semantic-segmentation (51 vs 50) pytorch-semantic-segmentation and Fast-SCNN-pytorch (51 vs 50) pytorch-semantic-segmentation and pytorch_tiramisu (51 vs 49)