UnReflectAnything: RGB-Only Highlight Removal by Rendering Synthetic Specular Supervision
By: Alberto Rota , Mert Kiray , Mert Asim Karaoglu and more
Potential Business Impact:
Removes shiny spots from pictures, showing true colors.
Specular highlights distort appearance, obscure texture, and hinder geometric reasoning in both natural and surgical imagery. We present UnReflectAnything, an RGB-only framework that removes highlights from a single image by predicting a highlight map together with a reflection-free diffuse reconstruction. The model uses a frozen vision transformer encoder to extract multi-scale features, a lightweight head to localize specular regions, and a token-level inpainting module that restores corrupted feature patches before producing the final diffuse image. To overcome the lack of paired supervision, we introduce a Virtual Highlight Synthesis pipeline that renders physically plausible specularities using monocular geometry, Fresnel-aware shading, and randomized lighting which enables training on arbitrary RGB images with correct geometric structure. UnReflectAnything generalizes across natural and surgical domains where non-Lambertian surfaces and non-uniform lighting create severe highlights and it achieves competitive performance with state-of-the-art results on several benchmarks. Project Page: https://alberto-rota.github.io/UnReflectAnything/
Similar Papers
Reflection Removal through Efficient Adaptation of Diffusion Transformers
CV and Pattern Recognition
Cleans up blurry photos by removing reflections.
Shift-Window Meets Dual Attention: A Multi-Model Architecture for Specular Highlight Removal
CV and Pattern Recognition
Removes shiny spots from pictures for clearer images.
IntrinsicReal: Adapting IntrinsicAnything from Synthetic to Real Objects
Graphics
Makes computer pictures show true colors better.