计算机应用

作者：廖光锴, 张正, 宋治国

单位：1.吉首大学 信息科学与工程学院，湖南 吉首 416000；2.吉首大学 物理与机电工程学院，湖南 吉首 416000

关键词：车辆重识别,通道注意力,像素注意力,小波变换,卷积神经网络

基金：国家自然科学基金资助项目(32060238)

针对现有的基于卷积神经网络（CNN）的车辆重识别方法所提取的特征表达力不足的问题，提出一种基于小波特征与注意力机制相结合的车辆重识别方法。首先，将单层小波模块嵌入到卷积模块中代替池化层进行下采样，减少细粒度特征的丢失；其次，结合通道注意力（CA）机制和像素注意力（PA）机制提出一种新的局部注意力模块——特征提取模块（FEM）嵌入到卷积网络中，对关键信息进行加权强化。在VeRi数据集上与基准残差网络ResNet-50、ResNet-101进行对比。实验结果表明，在ResNet-50中增加小波变换层数能提高平均精度均值（mAP）；在消融实验中，虽然ResNet-50+离散小波变换（DWT）比ResNet-101的mAP降低了0.25个百分点，但是其参数量和计算复杂度都比ResNet-101低，且mAP、Rank-1和Rank-5均比单独的ResNet-50高，说明该模型在车辆重识别中能够有效提高车辆检索精度。

来源：2022年第6期

《计算机应用》期刊编辑部

查看计算机应用杂志2022年第6期