基于注意力和视图信息的单图三维模型检索

摘要

为提取有效特征描述符，减小图像和三维模型巨大差异，提出一种基于注意力和视图信息的方法。该方法在模型特征提取模块引入空间注意力机制，提高模型特征描述符的有效性；将三维模型二维视图引入到查询图像特征学习过程中，缩小图像域与模型域域间差异。在Pix3D、 Comp Cars、 Stanford Cars 3个代表性基准数据集进行试验，结果表明检索精度较现有经典方法提高约5%。所提出方法能够使单幅图像有效检索相似三维模型，提高检索准确率。

关键词

三维模型 / 单幅图像 / 空间注意力 / 基于单图的三维模型检索 / 域间差异

基金资助

国家自然科学基金资助项目(62102163); 山东省高等学校青年创新团队发展计划资助项目; 山东省科技型中小企业创新能力提升工程资助项目(2023TSGC0244)