Computational Imaging Diffusion Diffusion Models

Diffusion models applied to specialized imaging tasks including medical image segmentation/reconstruction, computational imaging (CT, spectral imaging, RAW), and domain restoration (deconvolution, rolling shutter). Does NOT include general image generation, super-resolution as primary task, or text-conditional applications.

There are 116 computational imaging diffusion models tracked. The highest-rated is cswry/SeeSR at 48/100 with 618 stars.

Get all 116 projects as JSON

curl "https://pt-edge.onrender.com/api/v1/datasets/quality?domain=diffusion&subcategory=computational-imaging-diffusion&limit=20"

Open to everyone — 100 requests/day, no key needed. Get a free key for 1,000/day.

#	Model	Score	Tier	Stars	Language
1	cswry/SeeSR [CVPR2024] SeeSR: Towards Semantics-Aware Real-World Image Super-Resolution	48	Emerging	618	Python
2	JJLibra/SALAD-Pan 🤗 Official implementation for "SALAD-Pan: Sensor-Agnostic Latent Adaptive...	48	Emerging	33	Python
3	open-mmlab/mmgeneration MMGeneration is a powerful toolkit for generative models, based on PyTorch and MMCV.	47	Emerging	2,014	Python
4	Janspiry/Image-Super-Resolution-via-Iterative-Refinement Unofficial implementation of Image Super-Resolution via Iterative Refinement...	47	Emerging	3,910	Python
5	hanjq17/Spectrum [CVPR 2026] Adaptive Spectral Feature Forecasting for Diffusion Sampling Acceleration	44	Emerging	95	Python
6	IceClear/StableSR [IJCV2024] Exploiting Diffusion Prior for Real-World Image Super-Resolution	44	Emerging	2,641	Python
7	Kevin-thu/DiffMorpher Official Code for DiffMorpher: Unleashing the Capability of Diffusion Models...	43	Emerging	506	Python
8	mazurowski-lab/segmentation-guided-diffusion [MICCAI 2024] Easy diffusion models (optionally with segmentation guidance)...	42	Emerging	211	Python
9	aliasgharkhani/SLiMe 1-shot image segmentation using Stable Diffusion	42	Emerging	143	Python
10	pfriedri/wdm-3d [DGM4MICCAI'24] PyTorch implementation for "WDM: 3D Wavelet Diffusion Models...	42	Emerging	130	Python
11	atfortes/BokehDiffusion [SIGGRAPH Asia 2025] Official code for "Bokeh Diffusion: Defocus Blur...	41	Emerging	120	Python
12	Qiukunpeng/Siamese-Diffusion [CVPR 2025] Noise-Consistent Siamese-Diffusion for Medical Image Synthesis...	41	Emerging	87	Python
13	Tsingularity/dift [NeurIPS'23] Emergent Correspondence from Image Diffusion	41	Emerging	754	Python
14	ChenWu98/cycle-diffusion [ICCV 2023] A latent space for stochastic diffusion models	40	Emerging	652	Python
15	google/diffseg DiffSeg is an unsupervised zero-shot segmentation method using attention...	40	Emerging	330	Jupyter Notebook
16	mvrl/GeoSynth A PyTorch implementation of "GeoSynth: Contextually-Aware High-Resolution...	40	Emerging	111	Python
17	lin-tianyu/Stable-Diffusion-Seg [MICCAI 2024] Codebase for "Stable Diffusion Segmentation for Biomedical...	39	Emerging	112	Python
18	Con6924/SPM Official implementation of paper "One-dimensional Adapter to Rule Them All:...	39	Emerging	152	Python
19	lzzcd001/MeshDiffusion Official implementation of "MeshDiffusion: Score-based Generative 3D Mesh...	39	Emerging	828	Python
20	LarsDoorenbos/ccdm-stochastic-segmentation Repository for "Stochastic Segmentation with Conditional Categorical...	39	Emerging	52	Python
21	Wuziyi616/SlotDiffusion Code release for NeurIPS 2023 paper SlotDiffusion: Object-centric Learning...	38	Emerging	94	Python
22	mtkresearch/shortest-path-diffusion Official code for the paper "Image generation with shortest path diffusion"...	37	Emerging	24	Python
23	arplaboratory/ThermalGen [NeurIPS 2025] Official repository for "ThermalGen: Style-Disentangled...	37	Emerging	41	Python
24	pfriedri/pcdiff-implant [MICCAI'23] PyTorch implementation for "Point Cloud Diffusion Models for...	37	Emerging	50	Python
25	XinyuYanTJU/LawDIS [ICCV'2025] LawDIS: Language-Window-based Controllable Dichotomous Image Segmentation	36	Emerging	53	Python
26	JIA-Lab-research/DiffComplete Official Codebase of "DiffComplete: Diffusion-based Generative 3D Shape Completion"	36	Emerging	126	Python
27	thu-nics/ViDiT-Q [ICLR'25] ViDiT-Q: Efficient and Accurate Quantization of Diffusion...	36	Emerging	153	Python
28	PaulCouairon/DiffCut [NeurIPS 2024] Official code for DiffCut: Catalyzing Zero-Shot Semantic...	36	Emerging	50	Jupyter Notebook
29	lehduong/OneDiffusion Official implementation of OneDiffusion paper (CVPR 2025)	36	Emerging	665	Python
30	nauyihsnehs/IllumiDiff IllumiDiff: Indoor Illumination Estimation from a Single Image with...	36	Emerging	14	Python
31	fnzhan/EMLight [AAAI 2021] EMLight: Lighting Estimation via Spherical Distribution...	36	Emerging	173	Python
32	tsunghan-wu/SLD 🔥 [CVPR2024] Official implementation of "Self-correcting LLM-controlled...	36	Emerging	187	Python
33	SS47816/DriveSceneGen [RA-L 2024] DriveSceneGen: Generating Diverse and Realistic Driving...	35	Emerging	55	Python
34	ali-vilab/FreeScale [ICCV 2025] Code for FreeScale, a tuning-free method for higher-resolution...	35	Emerging	149	Python
35	vivoCameraResearch/SDMatte Official code for our ICCV2025 paper "SDMatte: Grafting Diffusion Models for...	35	Emerging	178	Python
36	lhf-labs/tm-dataset Code repository for our paper presenting the L3D dataset.	35	Emerging	24	Python
37	Malga-Vision/DiffusionXray-FewShot-LandmarkDetection Official PyTorch implementation of the paper: "Self-supervised pre-training...	35	Emerging	4	Python
38	dcodaaug/dcoda D-CODA: Diffusion for Coordinated Dual-Arm Data Augmentation (CoRL 2025)	35	Emerging	1	Python
39	jwliao1209/DiffQRCoder 🔥 [WACV'25] Official implementation of "DiffQRCoder: Diffusion-based...	35	Emerging	43	Python
40	mvrl/VectorSynth A PyTorch implementation of "VectorSynth: Fine-Grained Satellite Image...	34	Emerging	15	Python
41	ModelTC/TFMQ-DM [CVPR 2024 Highlight & TPAMI 2025] This is the official PyTorch...	34	Emerging	109	Jupyter Notebook
42	HealthX-Lab/DiffusionSynCTSeg [MICCAI2024] Official implementation of "CT-Based Brain Ventricle...	34	Emerging	10	Python
43	MediaBrain-SJTU/LED [CVPR2023] Leapfrog Diffusion Model for Stochastic Trajectory Prediction	34	Emerging	208	Jupyter Notebook
44	Zongliang-Wu/LADE-DUN [ECCV'24 - Best Paper Award Candidate & Oral] Latent Diffusion Prior...	34	Emerging	48	Python
45	VinAIResearch/Dataset-Diffusion Dataset Diffusion: Diffusion-based Synthetic Data Generation for Pixel-Level...	33	Emerging	128	Jupyter Notebook
46	AlonzoLeeeooo/LCDG The official code implementation of "LaCon: Late-Constraint Diffusion for...	33	Emerging	37	Python
47	metadriverse/SimGen Simulator-conditioned Driving Scene Generation	33	Emerging	134	Python
48	zblasingame/Greedy-DiM [IJCB 2024] Official PyTorch implementation for the paper "Greedy-DiM:...	33	Emerging	4	Python
49	jaychempan/EarthSynth Official Code for “EarthSynth: Generating Informative Earth Observation with...	33	Emerging	54	Python
50	urban-mobility-generation/Cardiff Leveraging the Spatial Hierarchy: Coarse-to-fine Trajectory Generation via...	32	Emerging	12	Jupyter Notebook
51	pmh9960/SphereDiff Official PyTorch implementation of "SphereDiff: Tuning-free Omnidirectional...	32	Emerging	50	Python
52	SonyResearch/RAW-Diffusion [WACV 2025] Official implementation of "RAW-Diffusion: RGB-Guided Diffusion...	32	Emerging	60	Python
53	xmed-lab/NEURONS [ICCV 2025] Neurons: Emulating the Human Visual Cortex Improves Fidelity and...	31	Emerging	26	Python
54	45-12/HNDSR 🌍 Enhance satellite images with HNDSR, a hybrid model that merges Neural...	31	Emerging	2	Jupyter Notebook
55	Jiahao000/MosaicFusion [IJCV 2024] MosaicFusion: Diffusion Models as Data Augmenters for Large...	31	Emerging	128	Python
56	shermanlian/spatial-entropy-loss Equipping Diffusion Models with Differentiable Spatial Entropy for Low-Light...	31	Emerging	16	Python
57	shercoo/RGDiffSR The official pytorch implementation of Paper: RECOGNITION-GUIDED DIFFUSION...	30	Emerging	6	Python
58	Gen-Verse/Diffusion-Sharpening Diffusion-Sharpening: Fine-tuning Diffusion Models with Denoising Trajectory...	30	Emerging	70	Python
59	G-U-N/consolver [CVPR 2026] Unofficial Implementation of "Image Diffusion Preview with...	30	Emerging	30	Python
60	HaibinLai/CrashSimGen Generating Safety-Critical Scenarios with Diffusion Models for self-Driving	30	Emerging	6	Jupyter Notebook
61	dongzhuoyao/self-guided-diffusion-models An official implementation of CVPR 2023 "Self-Guided Diffusion Models"	30	Emerging	27	Python
62	CUHK-AIM-Group/CLIFF [ECCV' 24 Oral] CLIFF: Continual Latent Diffusion for Open-Vocabulary Object...	29	Experimental	29	Python
63	ali-vilab/DiffDoctor [ICCV 2025] DiffDoctor: Diagnosing Image Diffusion Models Before Treating	29	Experimental	41	Python
64	georgosgeorgos/few-shot-diffusion-models Few-Shot Diffusion Models	29	Experimental	115	Python
65	sanghviyashiitb/kernel-diff-eccv24 Official Code for "Kernel Diffusion: An Alternate Approach to Blind...	28	Experimental	10	Python
66	lhaippp/RS-Diffusion [AAAI2025] Single Image Rolling Shutter Removal with Diffusion Models	28	Experimental	8	Python
67	lingyzhu0101/Diffusion_Image_Enhancement [TCSVT'25] Low-Light Image Enhancement via Diffusion Models with Semantic...	28	Experimental	6	Python
68	AlexMaOLS/EluCD Elucidating The Design Space of Classifier-Guided Diffusion Generation	27	Experimental	33	Python
69	EnVision-Research/DDSM Denoising Diffusion Step-aware Models (ICLR2024)	27	Experimental	62	Python
70	tangkai-RS/DreamCD [JAG 2026] DreamCD: A change-label-free framework for change detection via a...	27	Experimental	22	Python
71	thu-nics/MixDQ [ECCV24] MixDQ: Memory-Efficient Few-Step Text-to-Image Diffusion Models...	26	Experimental	49	Python
72	FengheTan9/Multi-Level-Global-Context-Cross-Consistency Official Pytorch Code base for "Multi-Level Global Context Cross Consistency...	26	Experimental	39	Python
73	xmed-lab/CvG-Diff [MICCAI 2025 Spotlight] Cross-view Generalized Diffusion Model for...	26	Experimental	10	Python
74	GATECH-EIC/Early-Bird-Diffusion [CVPR 2025] "Early-Bird Diffusion: Investigating and Leveraging...	25	Experimental	3	Python
75	brAIn-science/DeepSSIM [WACV, 2026] Official implementation of "A Novel Metric for Detecting...	25	Experimental	7	Python
76	TiankaiHang/Min-SNR-Diffusion-Training [ICCV 2023] Efficient Diffusion Training via Min-SNR Weighting Strategy	25	Experimental	268	Python
77	humansensinglab/AGenDA [ICCV 2025] Adapting Vehicle Detectors for Aerial Imagery to Unseen Domains...	25	Experimental	11	Python
78	YupeiLin2388/Diff-Mosaic [TGRS 2024]Diff-Mosaic: Augmenting Realistic Representations in Infrared...	24	Experimental	20	Python
79	alimd94/DiQP [WACV 2025] "Reversing the Damage: A QP-Aware Transformer-Diffusion Approach...	24	Experimental	19	Jupyter Notebook
80	xmed-lab/GenericSSL NeurIPS 2023: Towards Generic Semi-Supervised Framework for Volumetric...	24	Experimental	111	Python
81	EusdenLin/HDR-Reconstruction-Boosting a training-free approach that enhances existing indirect HDR reconstruction...	24	Experimental	2	Python
82	ZSDM-HDR/Zero-Shot-Diffusion-HDR [CVPR2024 Highlight Poster] Zero-Shot Structure-Preserving Diffusion Model...	23	Experimental	30	Python
83	choiyoungeunn/ReCo-Diff ReCo-Diff: Residual-Conditioned Deterministic Sampling for Cold Diffusion in...	23	Experimental	2	Python
84	chinmay5/semantically_consistent_graph_generation Official code for the MICCAI 2025 paper "Semantically Consistent Discrete...	23	Experimental	19	Python
85	covisionlab/diffusion_labeling Official implementation of "Bounding Box-Guided Diffusion for Synthesizing...	22	Experimental	10	Python
86	yjwong1999/RSGuidedDiffusion [IEEE BigData 2024] Cross-City Building Instance Segmentation: From More...	22	Experimental	8	Python
87	DM4CT/DM4CT Code for ICLR2026 paper - DM4CT: Benchmarking Diffusion Models for Computed...	22	Experimental	14	Python
88	sjtuplayer/few-shot-diffusion [ICCV 2023] Phasic Content Fusing Diffusion Model with Directional...	22	Experimental	67	Python
89	ash1ra/ResShift ResShift (Residual Shifting) model for SISR (Single Image Super-Resolution)...	22	Experimental	—	—
90	gudaochangsheng/WaDi [CVPR 2026] Official PyTorch implementation of WaDi: Weight Direction-aware...	21	Experimental	9	Python
91	lzzcd001/nabla-gfn Official Implementation of Nabla-GFlowNet (ICLR 2025)	21	Experimental	28	Python
92	FouierL/Di-Fusion [ICLR 2025] Official code of the paper "Self-Supervised Diffusion MRI...	21	Experimental	30	Python
93	GATECH-EIC/PostDiff [ICCV 2025] Fewer Denoising Steps or Cheaper Per-Step Inference: Towards...	21	Experimental	5	Python
94	wlsdzyzl/meshage High-Fidelity Medical Shape Generation via Skeletal Latent Diffusion	21	Experimental	5	Python
95	chinmay5/vessel_diffuse Official code for the MICCAI 2024 paper "3D Vessel Graph Generation Using...	21	Experimental	24	Python
96	FouierL/EquS Official Code of the paper “Equivariant Sampling for Improving Diffusion...	21	Experimental	19	Python
97	aimagelab/MAD Official PyTorch implementation for "Merging and Splitting Diffusion Paths...	21	Experimental	14	Python
98	vivoCameraResearch/AdaRefSR AdaRefSR is a novel reference-based one-step diffusion super-resolution...	20	Experimental	28	Python
99	JohnDreamer/DualCycleDiffsion The official repository of Dual-Cycle Diffusion (TCSVT 2023)	20	Experimental	6	Python
100	icon-lab/SSDiffRecon Official implementation of Self-Supervised Diffusion Model for MRI Reconstruction	20	Experimental	6	Python
101	morganm94/MultiCorrupt A Multi-Modal Robustness Dataset and Benchmark of LiDAR-Camera Fusion for 3D...	20	Experimental	5	Python
102	sinAshish/TreeDiffusion [MICCAI 2024] TrIND: Representing Anatomical Trees by Denoising Diffusion of...	20	Experimental	5	Python
103	steevelaquitaine/ilocal Software to decompose the mutual information I(X;R) between high-dimensional...	19	Experimental	2	Jupyter Notebook
104	basiralab/Dif-GSR Diffusion-based Graph Super-resolution	19	Experimental	4	Python
105	TReg-inverse/TReg Official repository for "Regularization by Texts for Latent Diffusion...	19	Experimental	17	Python
106	Ephemeral182/ECCV24_T3-DiffWeather [ECCV‘24] Teaching Tailored to Talent: Adverse Weather Restoration via...	19	Experimental	45	Python
107	sfu-mial/TreeDiffusion Official Code for "Representing Anatomical Trees by Denoising Diffusion of...	18	Experimental	2	Python
108	DreamSampler/dream-sampler The official repository for DreamSampler (ECCV24)	18	Experimental	37	Python
109	Likalto4/CEM-Detect Implementation of the Deep Breath MICCAI 2024 workshop paper "ELK: Enhanced...	18	Experimental	2	Jupyter Notebook
110	3587jjh/LSRNA Latent Space Super-Resolution for Higher-Resolution Image Generation with...	18	Experimental	46	Python
111	0606zt/TwinDiffusion [ECAI 2024] Official code for "TwinDiffusion: Enhancing Coherence and...	15	Experimental	29	Python
112	DominicVinxander/DSFNet 🌟 Learn to factor scenarios for improved multi-scenario route ranking with...	14	Experimental	—	Python
113	Yux1angJi/DIFF [ICASSP 2025] Diffusion Features to Bridge Domain Gap for Semantic Segmentation	14	Experimental	18	Python
114	tarhanefe/dds-relight This repository contains the code for the EPFL semester project conducted in...	11	Experimental	1	Python
115	alcros33/ThermalSuperResolution [CVPR 2024] Exploring the usage of diffusion models for thermal image...	11	Experimental	3	Python
116	sfu-mial/SLiMe Segment-Like-Me: 1-shot image segmentation using Stable Diffusion	10	Experimental	2	Python