基于三维注意力机制的车辆重识别算法

来源：计算机测量与控制杂志2022年第7期北京时间：

作者：方彦策,张宏江,谢雨成,刘培顺

单位：中国海洋大学 信息科学与工程学部

摘要：为解决套牌车识别难度大的问题,利用深度学习的技术,基于ResNet-50,结合通道注意力机制和位置注意力机制,设计了一种三维注意力机制对近似车辆进行精确识别;当前大部分注意力算法都关注于一维的通道注意力和二维的位置注意力,而处理的图像是三维的,这些注意力机制不能将注意力集中在所有需要关注的区域,造成部分关键信息遗失;该三维注意力机制在多种视觉任务下均有很好的效果,在Cifar100数据集上,相比SENet有1.12%的提升,在PKU VehicleID数据集上,相比SENet平均有2%的提升。

关键词：注意力机制;深度学习;套牌车识别;交通管理;

基金资助：国家重点研发计划