计算机与现代化 ›› 2022, Vol. 0 ›› Issue (03): 23-29.

• 人工智能 • 上一篇    下一篇

基于非局部注意力和局部特征的车辆重识别算法

  

  1. (1.河南工业大学信息科学与工程学院,河南郑州450001;2.河南省粮食信息处理国际联合实验室,河南郑州450001)
  • 出版日期:2022-04-29 发布日期:2022-04-29
  • 作者简介:万冬厚(1997—),男,山东德州人,硕士研究生,研究方向:数字图像处理,车辆重识别,E-mail: dreamer_wan@qq.com; 张德贤(1961—),男,河南新密人,教授,博士,研究方向:智能信息技术,E-mail: zdx@haut.edu.cn; 邓淼磊(1977—),男,河南南阳人,教授,博士,研究方向:信息安全,物联网技术,E-mail: dmlei2003@163.com。

Vehicle Re-identification Method Based on Non-local Attention and Local Features

  1. (1. School of Information Science and Engineering, Henan University of Technology, Zhengzhou 450001, China; 
    2. Henan International Joint Laboratory of Grain Information Processing, Zhengzhou 450001, China)
  • Online:2022-04-29 Published:2022-04-29

摘要: 车辆重识别是指从不同的摄像机来重新识别出同一辆车。车辆重识别非常容易受到车辆角度以及光照等其他因素的影响,是一项非常有挑战性的任务。许多车辆重识别方法都过分关注车辆全局特征,而忽略了车辆图像的局部有分辨力的特征,造成了车辆重识别精度不高的问题。针对这一问题,本文提出一种整合非局部注意力的和多尺度特征的车辆重识别方法,使用注意力机制获取车辆显著特征,并融合多尺度特征从而提高车辆重识别的检索精度。首先,使用骨干特征提取网络与注意力模块获取车辆的显著性细粒度特征。然后,将特征分为多个分支进行度量学习,分别学习车辆的局部与全局特征,将全局特征与细粒度的局部特征融合,构建车辆重识别的特征。最后,利用该方法提取不同车辆的特征,计算不同车辆的相似度,从而判断是否具有相同的身份。实验结果表明本文提出的车辆重识别算法具有更高的精度。

关键词: 非局部注意力, 局部特征, 车辆重识别, 神经网络

Abstract: Vehicle re-identification refers to re-identifying the same vehicle from different cameras. The result of vehicle re-identification is easily affected by other factors such as vehicle angle and illumination, which is a very challenging task. Many vehicle re-identification methods pay too much attention to the global features of the vehicle, but ignore the local resolution features of the vehicle image, which result in the problem of low accuracy of vehicle re-recognition. To solve this problem, this paper proposes a vehicle re-identification method integrating non-local attention and multi-scale features. The attention mechanism is used to obtain vehicle salient features and integrate multi-scale features, so as to improve the retrieval accuracy of vehicle re-identification. Firstly, the backbone feature extraction network and attention module are used to obtain the significant fine-grained features of vehicles. Then, the feature is divided into multiple branches for metric learning. The local and global features of vehicles are learned respectively, and the global features and fine-grained local features are fused to construct the features of vehicle re-identification. Finally, this method is used to extract the characteristics of different vehicles and calculate the similarity of different vehicles and judge whether they have the same identity. The experimental results show that the vehicle re-identification algorithm using attention mechanism and local features has higher accuracy.

Key words: non-local attention, local features, vehicle re-identification, neural network