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日
论文浅尝 | GEOM-GCN: Geometric Graph Convolutional Networks
开放知识图谱
14+阅读 · 2020年4月8日
【NeurIPS2019】图变换网络:Graph Transformer Network
国家自然科学基金
1+阅读 · 2015年12月31日
国家自然科学基金
3+阅读 · 2015年12月31日
国家自然科学基金
0+阅读 · 2015年12月31日
国家自然科学基金
0+阅读 · 2015年12月31日
国家自然科学基金
3+阅读 · 2015年12月31日
VIP会员
相关基金
国家自然科学基金
1+阅读 · 2015年12月31日
国家自然科学基金
3+阅读 · 2015年12月31日
国家自然科学基金
0+阅读 · 2015年12月31日
国家自然科学基金
0+阅读 · 2015年12月31日
国家自然科学基金
3+阅读 · 2015年12月31日
Top
微信扫码咨询专知VIP会员