[发明专利]基于多层次特征与注意力机制的行人再识别方法

权利要求书

1.一种基于多层次特征与注意力机制的行人再识别方法，其特征在于：包含数据预处理、注意力机制方法及多特征提取，其中所述数据预处理包括随机擦除和随机加噪的数据增强方法；所述注意力机制方法包括通道注意力机制和空间注意力机制两种注意力机制方法，所述多特征提取包括全局特征提取和局部特征提取两个分支特征提取方法。

2.根据权利要求1所述的基于多层次特征与注意力机制的行人再识别方法，其特征在于：在进行随机擦除和随机加噪处理时，对输入的数据集中50％的数据进行处理。

3.根据权利要求1所述的基于多层次特征与注意力机制的行人再识别方法，其特征在于：在数据预处理前准备图像样本集，所述图像样本集为采集的行人图像样本集及Market1501、CUHK03和DukeMTMC-ReID。

4.根据权利要求1所述的基于多层次特征与注意力机制的行人再识别方法，其特征在于：数据预处理完成后，将Resnet50网络作为骨干网络，并且在网络的第四层后添加空间注意力机制和通道注意力机制；所述空间注意力机制的作用是为了寻找网络中最重要的部位进行处理，所述通道注意力机制的作用是通过利用不同通道的重要程度从而有针对性地增强或抑制不同的通道。

5.根据权利要求1所述的基于多层次特征与注意力机制的行人再识别方法，其特征在于：对于全局分支，使用全局平局池化将特征映射转化为全局特征向量，用全局距离计算得到两幅图像的相似度。

6.根据权利要求1所述的基于多层次特征与注意力机制的行人再识别方法，其特征在于：对于局部分支，先采用水平池化将S3中得到的特征映射进行水平池化，得到池化后的特征映射，从而得到局部特征，对得到的局部特征进行局部距离测量，计算局部距离时采用最短距离的方法，得到两张图像的相似度。

7.根据权利要求5所述的基于多层次特征与注意力机制的行人再识别方法，其特征在于：还包括训练模型，所述训练模型使用传统的三元组损失函数计算损失，具体的，采用softmax损失函数和TriHard损失函数训练模型，总的损失函数为：

L＝L_ID+L_local+L_global

其中L_ID表示全局分支的softmax损失，L_global表示全局分支的TriHard损失，L_local表示局部分支的TriHard损失；采用总的损失函数训练模型，从而得到训练后的行人再识别模型。

8.一种基于多层次特征与注意力机制的行人再识别的模型，其特征在于：使用权利要求1-7任意一项所述的基于多层次特征与注意力机制的行人再识别方法构建得到。