基于Transformer的传统纹样子图检索方法

作者:彭宏*; 王炳烨; 高子惠
来源:中国传媒大学学报(自然科学版), 2022, 29(04): 8-18.
DOI:10.16196/j.cnki.issn.1673-4793.2022.04.001

摘要

针对如何有效、准确地检索传统纹样子图数据原图的问题,提出了一种基于Transformer的传统纹样子图检索算法。首先构建传统纹样子图数据集,其次利用卷积神经网络提取多层次的特征图进行融合。针对数据库中的图像利用Transformer生成预测框,将预测框映射回融合特征图提取局部与全局特征图后利用特征聚合算法聚合为全局与局部特征向量;针对查询子图,仅利用特征聚合算法对融合特征图进行聚合为全局向量。之后将查询子图的特征向量与数据库图像的特征向量分别进行相似度计算,排序后得到检索结果。对比实验证明本论文子图检索模型的有效性。

全文