Semantic Segmentation Techniques ML Frameworks

Research frameworks and methods for semantic segmentation using weak supervision, zero-shot learning, prototype-based approaches, and unsupervised feature learning. Does NOT include instance segmentation, panoptic segmentation, general video processing, or foreground/background separation as primary focus.

There are 78 semantic segmentation techniques frameworks tracked. 1 score above 70 (verified tier). The highest-rated is deepinv/deepinv at 73/100 with 674 stars. 1 of the top 10 are actively maintained.

Get all 78 projects as JSON

curl "https://pt-edge.onrender.com/api/v1/datasets/quality?domain=ml-frameworks&subcategory=semantic-segmentation-techniques&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	deepinv/deepinv DeepInverse: a PyTorch library for solving imaging inverse problems using...	73	Verified	674	Python
2	yjxiong/tsn-pytorch Temporal Segment Networks (TSN) in PyTorch	44	Emerging	1,085	Python
3	mhamilton723/STEGO Unsupervised Semantic Segmentation by Distilling Feature Correspondences	44	Emerging	785	Jupyter Notebook
4	fidler-lab/polyrnn-pp Inference Code for Polygon-RNN++ (CVPR 2018)	44	Emerging	734	Jupyter Notebook
5	oeway/pytorch-deform-conv PyTorch implementation of Deformable Convolution	42	Emerging	911	Python
6	VSainteuf/utae-paps PyTorch implementation of U-TAE and PaPs for satellite image time series...	42	Emerging	197	Jupyter Notebook
7	pyxu-org/pyxu Modular and scalable computational imaging in Python with GPU/out-of-core computing.	42	Emerging	140	Python
8	xvjiarui/GCNet GCNet: Non-local Networks Meet Squeeze-Excitation Networks and Beyond	41	Emerging	1,220	Python
9	philferriere/tfwss Weakly Supervised Segmentation with Tensorflow. Implements instance...	41	Emerging	221	Jupyter Notebook
10	nqanh/affordance-net AffordanceNet - Multiclass Instance Segmentation Framework - ICRA 2018	41	Emerging	134	Jupyter Notebook
11	NVlabs/pacnet Pixel-Adaptive Convolutional Neural Networks (CVPR '19)	40	Emerging	518	Python
12	twke18/SPML Universal Weakly Supervised Segmentation by Pixel-to-Segment Contrastive Learning	38	Emerging	105	Python
13	cheerss/CrossFormer The official code for the paper: https://openreview.net/forum?id=_PHymLIxuI	37	Emerging	401	Python
14	Haochen-Wang409/U2PL [CVPR'22 & IJCV'24] Semi-Supervised Semantic Segmentation Using Unreliable...	37	Emerging	474	Python
15	kqwang/phase-recovery Resources for phase recovery (also called phase imaging, phase retrieval, or...	37	Emerging	1,121	—
16	mmontielpz/jetseg NeurIPS 2023	36	Emerging	26	Jupyter Notebook
17	wasidennis/AdaptSegNet Learning to Adapt Structured Output Space for Semantic Segmentation, CVPR...	36	Emerging	861	Python
18	YichuXu/DSFormer [Neural Networks 2025] Dual Selective Fusion Transformer Network for...	36	Emerging	94	Python
19	developer0hye/PyTorch-Deformable-Convolution-v2 Don't feel pain to use Deformable Convolution	35	Emerging	343	Jupyter Notebook
20	hfslyc/AdvSemiSeg Adversarial Learning for Semi-supervised Semantic Segmentation, BMVC 2018	35	Emerging	507	Python
21	raoyongming/GFNet [NeurIPS 2021] [T-PAMI] Global Filter Networks for Image Classification	34	Emerging	504	Jupyter Notebook
22	kevin-ssy/FishNet Implementation code of the paper: FishNet: A Versatile Backbone for Image,...	34	Emerging	545	Python
23	ControlNet/MARLIN [CVPR] MARLIN: Masked Autoencoder for facial video Representation LearnINg	34	Emerging	261	Python
24	fidler-lab/polyrnn-pp-pytorch PyTorch training/tool code for Polygon-RNN++ (CVPR 2018)	33	Emerging	702	—
25	je-santos/ms_net The MultiScale Network for hierarchical regression (MS-Net) performs 3D...	33	Emerging	27	Jupyter Notebook
26	Kangningthu/ADELE Adaptive Early-Learning Correction for Segmentation from Noisy Annotations...	33	Emerging	89	Python
27	ouyangbaicai/MSI-DTrans Official PyTorch implementation of “MSI-DTrans: a multi-focus image fusion...	33	Emerging	6	Python
28	kqwang/computational-imaging Resources for computational imaging	32	Emerging	28	—
29	matciotola/Z-PNN Pansharpening by convolutional neural networks in the full resolution framework	31	Emerging	64	Python
30	yc-cui/Deep-Learning-Spatiotemporal-Fusion-Survey [Information Fusion 2026] A collection of deep learning models for remote...	31	Emerging	53	Python
31	nv-tlabs/STEAL STEAL - Learning Semantic Boundaries from Noisy Annotations (CVPR 2019)	31	Emerging	479	Jupyter Notebook
32	TonyLianLong/CrossMAE Official Implementation of the CrossMAE paper: Rethinking Patch Dependence...	30	Emerging	133	Python
33	SegDAC/SegDAC Official repository of SegDAC: Improving Visual Reinforcement Learning by...	30	Emerging	12	Jupyter Notebook
34	AmirMansurian/AICSD Adaptive Inter-Class Similarity Distillation for Semantic Segmentation (MTAP 2025)	30	Emerging	29	Python
35	zgr6010/HSI_SSFTT L. Sun, G. Zhao, Y. Zheng, and Z. Wu. "Spectral–Spatial Feature Tokenization...	29	Experimental	128	Python
36	daniel-scholz/mm-dinov2 Code for MM-DINOv2: Adapting Foundation Models for Multi-Modal Medical Image...	29	Experimental	19	Python
37	Laknath1996/DeepPhaseUnwrap This repository Introduces a joint convolutional and spatial...	29	Experimental	52	Python
38	ViTAE-Transformer/ViTAE-Transformer-Remote-Sensing A comprehensive list [SAMRS@NeurIPS'23, RVSA@TGRS'22, RSP@TGRS'22] of our...	29	Experimental	486	TeX
39	hynnsk/HP This is the official pytorch implementation of "Leveraging Hidden Positives...	28	Experimental	88	Python
40	HSG-AIML/SSLTransformerRS Code repository for "Self-supervised Vision Transformers for Land-cover...	28	Experimental	92	Python
41	jakhac/CSMAE Cross-Sensor Masked Autoencoder for Content Based Image Retrieval in Remote Sensing	28	Experimental	27	Python
42	dronefreak/JetNemotron-Extension Hybrid CNN + JetBlock vision models in PyTorch for segmentation, depth...	28	Experimental	5	Python
43	twke18/HSG Unsupervised Hierarchical Semantic Segmentation with Multiview...	28	Experimental	74	Python
44	Ni-Chen/3D-MB-HoloNet Code of "Model-based holographic network for 3D particle imaging", IEEE...	27	Experimental	7	MATLAB
45	yc-cui/PEMAE [TGRS 2024] PEMAE: Pixel-Wise Ensembled Masked Autoencoder for Multispectral...	27	Experimental	8	Python
46	tonysy/Deep-Feature-Flow-Segmentation Deep Feature Flow for Video Semantic Segmentation	27	Experimental	35	Python
47	lok-18/A2RNet AAAI 2025 \| A2RNet: Adversarial Attack Resilient Network for Robust Infrared...	26	Experimental	31	Python
48	DegangWang97/IEEE_JSTARS_NL2Net [JSTARS 2025 ESI Highly Cited Paper (TOP 1%)] Non-Local and Local...	26	Experimental	11	Python
49	shjo-april/RecurSeed_and_EdgePredictMix [Under Review] RecurSeed and EdgePredictMix: Single-stage learning is...	26	Experimental	44	Python
50	micahcarroll/uniMASK Codebase for "Uni[MASK]: Unified Inference in Sequential Decision Problems"	26	Experimental	57	Python
51	chen742/PiPa Official Implementation of PiPa: Pixel- and Patch-wise Self-supervised...	26	Experimental	100	Python
52	pashtari/deconver Official PyTorch Implementation of "Deconver: A Deconvolutional Network for...	25	Experimental	6	Python
53	AmirMansurian/AttnFD [WACV'26] Attention as Geometric Transformation: Revisiting Feature...	24	Experimental	42	Python
54	Genie-Kim/PintheMemory Pin the Memory (CVPR 2022): Official Project Webpage	24	Experimental	44	Python
55	reyjosephnieto/ph Computational Stability of Cubical Homology: Orthogonal Controls for Fault...	23	Experimental	1	Python
56	threedle/GeoCode GeoCode maps 3D shapes to a human-interpretable parameter space, allowing to...	23	Experimental	412	Python
57	alpinedev51/Hyperspectral-Deep-Deconvolution Project on using deep models for deconvolution of hyperspectral images with...	23	Experimental	2	TeX
58	mrk1992/robust_matching_network_on_remote_sensing_imagery_pytorch A Robust Matching Network for Gradually Estimating Geometric Transformation...	23	Experimental	33	Python
59	kqwang/DLPR Example code for data-driven and physics-driven deep learning phase recovery	23	Experimental	41	Python
60	DegangWang97/IEEE_GRSL_PUNNet [GRSL 2024] Global Feature-Injected Blind-Spot Network for Hyperspectral...	22	Experimental	7	Python
61	yasserben/CLOUDS [CVPR 2024] Official Implementation of Collaborating Foundation models for...	22	Experimental	76	Python
62	Talegqz/unsupervised_co_part_segmentation [ICML2021] Unsupervised Co-part Segmentation through Assembly	22	Experimental	45	Python
63	conradry/pixpro-with-weights Pixel Propagation for unsupervised visual representation learning	21	Experimental	11	Python
64	enter-i-username/MSNet [TGRS 2024] MSNet: Self-Supervised Multiscale Network with Enhanced...	20	Experimental	19	Python
65	DotWang/DCN-T [TIP 2023] The official repo for the paper "DCN-T: Dual Context Network with...	20	Experimental	47	Python
66	naver/unic PyTorch code and pretrained weights for the UNIC models.	20	Experimental	44	Python
67	Sense-X/UniNet Unified Architecture Search with Convolution, Transformer, and MLP (ECCV 2022)	20	Experimental	53	Python
68	vladan-stojnic/CMC-RSSR This repository contains the code and models from the paper ...	18	Experimental	37	Python
69	chaofanma/AttrSeg [NeurIPS 2023] "AttrSeg: Open-Vocabulary Semantic Segmentation via Attribute...	18	Experimental	3	—
70	antoinehabis/Deep-ContourFlow Unsupervised, one-shot, instance-based active contour/snake using deep...	18	Experimental	18	Python
71	licongguan/ILM-ASSL Iterative Loop Method Combining Active and Semi-Supervised Learning for...	16	Experimental	34	Python
72	laprf/SMN Code for Spectrum-driven Mixed-frequency Network for Hyperspectral Salient...	15	Experimental	16	Python
73	HSG-AIML/GDA Code repository for "Parameter Efficient Self-supervised Geospatial Domain...	14	Experimental	36	Python
74	OlegnaPT90/HiHolo ⚡ Streamline high-resolution X-ray phase contrast imaging with HiHolo's...	14	Experimental	—	C++
75	ruohaoguo/UniTR Official Implementation of "UniTR: A Unified TRansformer-based Framework for...	13	Experimental	6	Python
76	DanieleMorotti/Blind-Image-Separation This project has been developed as part of 'Deep Learning' course at the...	13	Experimental	5	Jupyter Notebook
77	apicis/aff-seg Official repository of the paper "Segmenting Object Affordances:...	13	Experimental	5	Python
78	measterpojo/Cross-Pseudo-Supervision-CPS-semi-supervised-segmentation Cross Pseudo Supervision (CPS) is a semi-supervised learning technique...	11	Experimental	—	Jupyter Notebook