Skip to content

[CVPR 2025] Official Implementation of "PolarFree: Polarization-based Reflection-Free Imaging"

Notifications You must be signed in to change notification settings

mdyao/PolarFree

Folders and files

NameName
Last commit message
Last commit date

Latest commit

Β 

History

8 Commits
Β 
Β 
Β 
Β 

Repository files navigation

πŸš€ PolarFree: Polarization-based Reflection-Free Imaging – [CVPR 2025]

🌟 A Cutting-Edge Solution and Dataset for Polarization-based Reflection-Free Imaging

Polarization-based Reflection and Refraction

Image source: ThinkLucid

πŸ”— Project Page | πŸ“„ Paper | πŸ“¦ [Dataset](Coming soon...)


πŸ“Œ Highlights

βœ… Large-Scale Dataset: PolaRGB includes 6,500 well-aligned RGB-polarization image pairs, 8Γ— larger than existing datasets.
βœ… Innovative Method: PolarFree leverages diffusion models to generate reflection-free priors for accurate reflection removal.
βœ… State-of-the-Art Performance: Outperforms existing methods by ~2dB in PSNR on challenging real-world scenarios.
βœ… Open Source: Code and dataset are freely available for research and development.

⏳ Timeline

  • βœ… 2025-03-23 - πŸ› οΈ Repository initialized with documentation.
  • βœ… 2025-03-23 - πŸ”— Project Page officially launched.
  • ⬜ TODO: πŸ“„ Paper available on arXiv.
  • ⬜ TODO: πŸš€ Provide core codebase and pre-trained models for evaluation.
  • ⬜ TODO: πŸ“¦ Release the full PolaRGB dataset with download links.
  • ⬜ TODO: πŸ“ Publish training code and instructions.

πŸ“– Overview

PolarFree addresses the challenging task of reflection removal using polarization cues and a novel diffusion-based approach. Key contributions include:

  • PolaRGB Dataset: A large-scale dataset with diverse indoor and outdoor scenes, providing RGB and polarization images.

Dataset Overview

  • Diffusion Model: Utilizes diffusion processes to generate reflection-free priors, enabling precise reflection removal and improved image clarity.
    Model Design

  • Superior Results: Extensive experiments on the PolaRGB dataset show that PolarFree outperforms existing methods by ~2dB in PSNR, achieving cleaner reflection removal and sharper image details.

  • Real-World Effectiveness: PolarFree demonstrates robust performance in real-world scenarios, such as museums and galleries, effectively reducing reflections while preserving fine details.


πŸš€ Installation

  1. Clone the repository:
    git clone https://github.com/mdyao/PolarFree.git
    cd PolarFree
    pip install -r requirements.txt
  2. Run the demo:
    python demo.py --input example.jpg --output result.jpg
    

πŸ“Š Results

PolarFree achieves superior performance compared to existing methods:

Results

πŸ“œ Citation

If you find this work useful, please cite:

@inproceedings{polarfree2025,
title={PolarFree: Polarization-based Reflection-Free Imaging},
author={Mingde Yao, Menglu Wang, King-Man Tam, Lingen Li, Tianfan Xue, Jinwei Gu},
booktitle={CVPR},
year={2025}
}

About

[CVPR 2025] Official Implementation of "PolarFree: Polarization-based Reflection-Free Imaging"

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published