Skip to the content.

Alberto Rota1, Mert Kiray2,3, Mert Asim Karaoglu4,2, Patrick Ruhkamp2, Elena De Momi1, Nassir Navab2,3, Benjamin Busam2,3
1Politecnico di Milano    2Technical University of Munich    3Munich Center for Machine Learning (MCML)    4ImFusion

Before After
Before After
Before After
Before After

Abstract

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

method overview