noakraicer/Guided-Lensless-Polarization-Imaging

Guided Lensless Polarization Imaging

Pretrained Stage II checkpoints for Guided Lensless Polarization Imaging (CVPR 2026 Findings).

Noa Kraicer, Erez Yosef, Raja Giryes · Tel Aviv University

Models

Checkpoint	Architecture	Guide	Description
grayscale_with_guide.pth	swinfusionSR_GRAYSCALE	RGB	3ch grayscale polarimetric (0°, 45°, 90°) with RGB guide
grayscale_no_guide.pth	swinfusionSR_GRAYSCALE	None	3ch grayscale polarimetric, no guide (FISTA + Transformer baseline)
color_with_guide.pth	swinfusionSRcolor	RGB	12ch color polarimetric (4 angles × 3 RGB) with RGB guide
color_no_guide.pth	swinfusionSRcolor	None	12ch color polarimetric, no guide (ablation)

Usage

git clone https://github.com/noa-kraicer/Guided-Lensless-Polarization-Imaging.git
cd Guided-Lensless-Polarization-Imaging
wget https://huggingface.co/noakraicer/Guided-Lensless-Polarization-Imaging/resolve/main/grayscale_with_guide.pth
# Set path/pretrained_netG in your config JSON, then:
python test_swin_final.py --opt options/test_RGBP.json

Citation

@misc{kraicer2026guidedlenslesspolarizationimaging,
  title         = {Guided Lensless Polarization Imaging},
  author        = {Noa Kraicer and Erez Yosef and Raja Giryes},
  year          = {2026},
  eprint        = {2603.27357},
  archivePrefix = {arXiv},
  primaryClass  = {eess.IV},
  url           = {https://arxiv.org/abs/2603.27357}
}