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/


翻译:镜面高光会扭曲外观、掩盖纹理并阻碍自然图像和手术图像中的几何推理。我们提出UnReflectAnything,一种仅基于RGB的框架,通过预测高光图与无反射的漫反射重建,从单张图像中去除高光。该模型使用冻结的视觉Transformer编码器提取多尺度特征,一个轻量级头部定位镜面区域,以及一个在生成最终漫反射图像前修复受损特征块的令牌级修复模块。为克服配对监督数据的缺乏,我们引入了虚拟高光合成流程,该流程利用单目几何、菲涅耳感知着色和随机化光照渲染物理上合理的镜面反射,从而能够在具有正确几何结构的任意RGB图像上进行训练。UnReflectAnything在自然和手术领域中均表现出良好的泛化能力,其中非朗伯表面和非均匀光照会产生严重高光,并在多个基准测试中取得了与最先进方法相竞争的性能。项目页面:https://alberto-rota.github.io/UnReflectAnything/

0
下载
关闭预览

相关内容

【CVPR2024】ViewDiff: 3D一致的图像生成与文本到图像模型
专知会员服务
30+阅读 · 2024年3月10日
AAAI 2022 | ProtGNN:自解释图神经网络
专知
10+阅读 · 2022年2月28日
【NeurIPS2019】图变换网络:Graph Transformer Network
国家自然科学基金
1+阅读 · 2015年12月31日
国家自然科学基金
0+阅读 · 2015年12月31日
国家自然科学基金
0+阅读 · 2015年12月31日
国家自然科学基金
3+阅读 · 2015年12月31日
国家自然科学基金
0+阅读 · 2014年12月31日
VIP会员
相关基金
国家自然科学基金
1+阅读 · 2015年12月31日
国家自然科学基金
0+阅读 · 2015年12月31日
国家自然科学基金
0+阅读 · 2015年12月31日
国家自然科学基金
3+阅读 · 2015年12月31日
国家自然科学基金
0+阅读 · 2014年12月31日
Top
微信扫码咨询专知VIP会员