arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2211.08089cs.CV
收藏

DeS3:利用ViT相似性的自适应注意力驱动自身阴影与软阴影去除方法

DeS3: Adaptive Attention-driven Self and Soft Shadow Removal using ViT Similarity

National University of Singapore(新加坡国立大学);Huawei International Pte Ltd(华为国际有限公司);Peng Cheng Laboratory(鹏城实验室)

Yeying Jin, Wei Ye, Wenhan Yang, Yuan Yuan, Robby T. Tan

AI总结:

针对单张图像中软阴影和自身阴影边界模糊难去除的问题,提出DeS3方法,通过采样阶段引入ViT相似性损失和自适应注意力,在多个数据集上实现优于现有最优方法的阴影去除效果。

摘要:

从单张图像中去除边界不清晰的软阴影和自身阴影仍然是一项具有挑战性的任务。自身阴影是指投射在物体自身上的阴影。现有大多数方法依赖二值阴影掩模,未考虑软阴影和自身阴影的模糊边界问题。本文提出DeS3方法,该方法基于自适应注意力和ViT(视觉Transformer)相似性,可去除硬阴影、软阴影和自身阴影。我们提出的新型ViT相似性损失利用预训练Vision Transformer提取的特征,该损失可引导反向采样过程恢复场景结构。我们的自适应注意力能够区分阴影区域与下方物体,也能区分阴影区域与投射阴影的物体,这一能力使DeS3即使在物体被阴影部分遮挡时,也能更好地恢复物体结构。与现有在训练阶段引入约束的方法不同,我们在采样阶段融入ViT相似性。我们的方法在SRD、AISTD、LRSS、USR和UIUC数据集上的表现优于现有最优方法,能够鲁棒地去除硬阴影、软阴影和自身阴影。具体而言,在LRSS数据集上,我们的方法在整图RMSE指标上比现有最优方法提升16%。我们的数据和代码可在以下链接获取:https://github.com/jinyeying/DeS3_Deshadow

英文摘要:

Removing soft and self shadows that lack clear boundaries from a single image is still challenging. Self shadows are shadows that are cast on the object itself. Most existing methods rely on binary shadow masks, without considering the ambiguous boundaries of soft and self shadows. In this paper, we present DeS3, a method that removes hard, soft and self shadows based on adaptive attention and ViT similarity. Our novel ViT similarity loss utilizes features extracted from a pre-trained Vision Transformer. This loss helps guide the reverse sampling towards recovering scene structures. Our adaptive attention is able to differentiate shadow regions from the underlying objects, as well as shadow regions from the object casting the shadow. This capability enables DeS3 to better recover the structures of objects even when they are partially occluded by shadows. Different from existing methods that rely on constraints during the training phase, we incorporate the ViT similarity during the sampling stage. Our method outperforms state-of-the-art methods on the SRD, AISTD, LRSS, USR and UIUC datasets, removing hard, soft, and self shadows robustly. Specifically, our method outperforms the SOTA method by 16\% of the RMSE of the whole image on the LRSS dataset. Our data and code is available at: \url{https://github.com/jinyeying/DeS3_Deshadow}

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑