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