基于多粒度特征分割的车辆重识别算法

Vehicle Re-identification Algorithm Based on Multi-granularity Feature Segmentation

针对车辆重识别任务中局部特征提取不充分和潜在显著性局部特征易被掩盖的问题,提出一种基于多粒度特征分割的算法。该算法采用可实现跨通道间信息交互的ResNeSt-50作为骨干网络提取初级特征,并将骨干网络复制成三个独立的分支,对输出的特征图分别沿纵向、横向和通道方向进行多粒度分割以提取到区分性局部特征。为进一步增强网络提取判别性特征信息的能力,又在ResNeSt-50的每个split-attention block中嵌入了空间注意力模块。研究结果表明:算法在VeRi-776数据集上的mAP、Rank-1、Rank-5指标分别达到85.92%、97.67%、98.53%;在VehicleID数据集的三个测试集上,Rank-1指标分别达到了88.36%、84.19%、78.89%,优于现有大部分主流算法,研究结果表明该算法具有先进性和有效性。

In order to solve the problem that local feature extraction was insufficient and potentially significant local features were easy to be concealed in vehicle re-identification task,an algorithm based on multi-granularity feature segmentation was proposed.In the proposed algorithm,ResNeSt-50 which could realize the information exchange between channels was adopted as the backbone network to extract primary features.And the backbone network was copied into three independent branches,and the output feature map was segmented by multi-granularity along the vertical,horizontal and channel directions to extract the differentiated local features.In order to further enhance the ability of the network to extract discriminant feature information,the spatial attention module was embedded in each split-attention block of ResNeSt-50.The research results show that The mAP,Rank-1 and Rank-5 indexes of the proposed algorithm on the VeRi-776 dataset reach 85.92%,97.67%and 98.53%respectively;on the three test sets of the VehicleID dataset,the Rank-1 index reaches 88.36%,84.19%and 78.89%respectively,which is better than that of most of the existing mainstream algorithms.The research results show that the proposed algorithm is advanced and effective.