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 111 computational imaging diffusion models tracked. The highest-rated is Janspiry/Image-Super-Resolution-via-Iterative-Refinement at 47/100 with 3,910 stars.

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

Comparisons in this category

VectorSynth and GeoSynth (33 vs 33)