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

基于对称变换器的镜像检测

Symmetry-Aware Transformer-based Mirror Detection

Harbin Institute of Technology(哈尔滨工业大学);City University of Hong Kong(香港城市大学)

Tianyu Huang, Bowen Dong, Jiaying Lin, Xiaohui Liu, Rynson W. H. Lau, Wangmeng Zuo

摘要:

镜像检测的目的是识别给定输入图像中的镜像区域。现有的工作主要集中在结合语义特征和结构特征来挖掘镜像区域和非镜像区域之间的特定关系,或者引入深度、手征等镜像属性来分析镜像的存在性。在这项工作中,我们观察到,一个真实的对象通常形成一个松散的对称关系,其相应的反射在镜子中,这是有益的区分镜子从真实的对象。基于这一观察,我们提出了一个双路径对称性感知的基于变换器的镜像检测网络(SATNet),它包括两个新的模块:对称性感知注意模块(SAAM)和对比度和融合解码器模块(CFDM)。具体来说,我们首先采用一个Transformer骨干来模拟图像中的全局信息聚合,在两条路径上提取多尺度特征。然后,我们饲料的高层次的双路径功能SAAM捕捉对称关系。最后,我们融合双路径特征,并使用CFDM逐步完善我们的预测图,以获得最终的镜像掩码。实验结果表明,SATNet优于RGB和RGB-D镜像检测方法在所有可用的镜像检测数据集。代码和训练模型可在https://github.com/tyhuang0428/SATNet上获得。

英文摘要:

Mirror detection aims to identify the mirror regions in the given input image. Existing works mainly focus on integrating the semantic features and structural features to mine specific relations between mirror and non-mirror regions, or introducing mirror properties like depth or chirality to help analyze the existence of mirrors. In this work, we observe that a real object typically forms a loose symmetry relationship with its corresponding reflection in the mirror, which is beneficial in distinguishing mirrors from real objects. Based on this observation, we propose a dual-path Symmetry-Aware Transformer-based mirror detection Network (SATNet), which includes two novel modules: Symmetry-Aware Attention Module (SAAM) and Contrast and Fusion Decoder Module (CFDM). Specifically, we first adopt a transformer backbone to model global information aggregation in images, extracting multi-scale features in two paths. We then feed the high-level dual-path features to SAAMs to capture the symmetry relations. Finally, we fuse the dual-path features and refine our prediction maps progressively with CFDMs to obtain the final mirror mask. Experimental results show that SATNet outperforms both RGB and RGB-D mirror detection methods on all available mirror detection datasets. Codes and trained models are available at: https://github.com/tyhuang0428/SATNet.

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

↑