File size: 1,948 Bytes
9719e2e 17b891f 9719e2e 28a57d7 6439fbe ccc7093 9719e2e bfa5e1b 9719e2e 23ba4f5 4435b90 9165ab2 9719e2e | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 | ---
license: cc-by-4.0
language:
- en
base_model:
- ByteDance/sd2.1-base-zsnr-laionaes5
pipeline_tag: image-text-to-image
tags:
- SPAD
- Photons
- Generative
- ISP
datasets:
- aRy4n/eXtreme-Deformable
- aRy4n/real-color-SPAD-indoor6
metrics:
- type: accuracy
split: test
task:
type: video-to-image
name: Burst Reconstruction
---
## gQIR: Generative Quanta Image Reconstruction
[Aryan Garg](https://aryan-garg.github.io/)<sup>1</sup>, [Sizhuo Ma](https://sizhuoma.netlify.app/)<sup>2</sup>, [Mohit Gupta](https://wisionlab.com/people/mohit-gupta/)<sup>1</sup>
<sup>1</sup> University of Wisconsin-Madison <sup>2</sup> Snap, Inc<br>
## All model weights are available here now!
| Color-Model Name | Stage | Bit Depth | 🤗 Download Link |
|:---|:---:|:---:|:---|
| qVAE | Stage 1 | 1-bit | [1965000.pt](https://huggingface.co/aRy4n/gQIR/resolve/main/1-bit/1965000.pt) |
| Adversarial Diffusion LoRA-UNet | Stage 2 | 1-bit | [state_dict.pth](https://huggingface.co/aRy4n/gQIR/resolve/main/1-bit/state_dict.pth) |
| qVAE | Stage 1 | 3-bit | [0105000.pt](https://huggingface.co/aRy4n/gQIR/resolve/main/0105000.pt) |
| Adversarial Diffusion LoRA-UNet | Stage 2 | 3-bit | [state_dict.pth](https://huggingface.co/aRy4n/gQIR/resolve/main/state_dict.pth) |
| FusionViT | Stage 3 | 3-bit | [fusion_vit_0050000.pt](https://huggingface.co/aRy4n/gQIR/resolve/main/fusion_vit_0050000.pt) |
Code at [github.com/Aryan-Garg/gQIR](https://github.com/Aryan-Garg/gQIR)
ArXiv Version: [arxiv.org/abs/2602.20417](https://arxiv.org/abs/2602.20417)
#### Cite Us:
```bibtex
@InProceedings{garg_2026_gqir,
author = {Garg, Aryan and Ma, Sizhuo and Gupta, Mohit},
title = {gQIR: Generative Quanta Image Reconstruction},
booktitle = {Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR)},
month = {June},
year = {2026},
}
``` |