Specular highlights on smooth surfaces obscure image details and degrade the performance of vision tasks such as segmentation and detection. Traditional RGB-based highlight removal methods rely heavily on supervised data, suffering from high acquisition costs and limited generalization. Polarization imaging provides complementary cues by separating specular reflections from diffuse components, offering a physical strategy to address this challenge. Inspired by this, we propose the Polarization-guided Specular Removal Diffusion Model (PSR-Diff), which integrates diffusion models with polarization priors. PSR-Diff extracts polarization cues via Fresnel theory to construct a pseudo-diffuse estimate and fuses it with RGB features through a cross-modal attention mechanism. To mitigate data scarcity and modality differences, we introduce a dual-module fusion strategy that leverages a pretrained RGB diffusion model while enabling polarization-guided learning. Experiments on polarization and RGB-only datasets show that PSR-Diff achieves superior highlight suppression, enhanced image quality, and robust performance under diverse lighting and materials.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

PSR-Diff: Polarization-Guided Diffusion Model for Single Image Specular Highlight Removal

  • Guobin Zhang,
  • Li Li,
  • Zhaojing Wang,
  • Qihang Wang,
  • Tao Peng,
  • Xinrong Hu

摘要

Specular highlights on smooth surfaces obscure image details and degrade the performance of vision tasks such as segmentation and detection. Traditional RGB-based highlight removal methods rely heavily on supervised data, suffering from high acquisition costs and limited generalization. Polarization imaging provides complementary cues by separating specular reflections from diffuse components, offering a physical strategy to address this challenge. Inspired by this, we propose the Polarization-guided Specular Removal Diffusion Model (PSR-Diff), which integrates diffusion models with polarization priors. PSR-Diff extracts polarization cues via Fresnel theory to construct a pseudo-diffuse estimate and fuses it with RGB features through a cross-modal attention mechanism. To mitigate data scarcity and modality differences, we introduce a dual-module fusion strategy that leverages a pretrained RGB diffusion model while enabling polarization-guided learning. Experiments on polarization and RGB-only datasets show that PSR-Diff achieves superior highlight suppression, enhanced image quality, and robust performance under diverse lighting and materials.