[发明专利]基于稀疏注意力网络的行人再辨识方法

摘要

本发明公开一种基于稀疏注意力网络的行人再辨识方法，首先通过短连接将浅层特征无损传输到深层特征；然后通过不断叠加的残差模块组成的主干残差网络来提取到图像主要卷及特征；接着通过主干残差网络中嵌入的归一化的压缩‑激励模块提取图像易被损失的细节特征；最后将上述得到的特征相乘，最后再加上第一部分所得的特征，输送进全连接层和分类回归层从而得到分类和回归结果。本发明的稀疏注意力网络可有效提取到数个行人再辨识数据集的行人照片细节特征。

主权项

1.基于稀疏注意力网络的行人再辨识方法，其特征是，包括如下步骤：步骤1、将已知的行人再辨识数据集中的图像分为训练集和测试集，并分别对训练集和测试集中的图像进行预处理；步骤2、对步骤1所得的训练集中的所有训练图像进行复制，分别得到原训练图像和复制训练图像；步骤3、对于步骤2所得的原训练图像，先将原训练图像送入卷积层提取图像的卷积特征，再将提取的卷积特征送入最大池化层提取图像的最大池化特征，后将提取的最大池化特征送入3个重复叠加的第一残差模块提取图像的第一残差卷积特征；步骤4、将步骤3所得的第一残差卷积特征送入第一归一化的压缩‑激励模块提取图像的第一注意力特征；步骤5、将步骤3所得的第一残差卷积特征和步骤4所得的第一注意力特征相乘，得到第一稀疏注意力特征；步骤6、利用步骤2所的复制训练图像与步骤5所得的第一稀疏注意力特征相加，得到第一阶段图像特征；步骤7、对步骤6所得的所有第一阶段图像特征进行复制，分别得到原第一阶段图像特征和复制第一阶段图像特征；步骤8、对于步骤7所得的原第一阶段图像特征特征送入4个重复叠加的第二残差模块提取图像的第二残差卷积特征；步骤9、将步骤8所得到第二残差卷积特征送入第二归一化的压缩‑激励模块提取图像的第二注意力特征；步骤10、将步骤8所得的第二残差卷积特征与步骤9所得的第二注意力特征相乘，得到第二稀疏注意力特征；步骤11、将步骤7所得的复制第一阶段图像特征与步骤10所得的第二稀疏注意力特征相加，得到第二阶段图像特征；步骤12、对步骤11所得的所有第二阶段图像特征进行复制，分别得到原第二阶段图像特征和复制第二阶段图像特征；步骤13、对于步骤12所得的原第二阶段图像特征特征送入6个重复叠加的第三残差模块提取图像的第三残差卷积特征；步骤14、将步骤13所得到第三残差卷积特征送入第三归一化的压缩‑激励模块提取图像的第三注意力特征；步骤15、将步骤13所得的第三残差卷积特征与步骤14所得的第三注意力特征相乘，得到第三稀疏注意力特征；步骤16、将步骤12所得的复制第二阶段图像特征与步骤15所得的第三稀疏注意力特征相加，得到第三阶段图像特征；步骤17、对步骤16所得的所有第三阶段图像特征进行复制，分别得到原第三阶段图像特征和复制第三阶段图像特征；步骤18、对于步骤17所得的原第三阶段图像特征特征送入3个重复叠加的第四残差模块提取图像的第四残差卷积特征；步骤19、将步骤18所得到第四残差卷积特征送入第四归一化的压缩‑激励模块提取图像的第四注意力特征；步骤20、将步骤18所得的第四残差卷积特征与步骤19所得的第四注意力特征相乘，得到第四稀疏注意力特征；步骤21、将步骤17所得的复制第三阶段图像特征与步骤20所得的第四稀疏注意力特征相加，得到第四阶段图像特征；步骤22、将步骤21所得的所有第四阶段图像特征送入平均池化层提取图像的平均池化特征；步骤23、将步骤22所得到的所有平均池化特征送入分类层，由此得到行人类别的预测模型；步骤24、利用步骤2所得的测试集中的所有测试图像对步骤23所得的行人类别的预测模型进行测试，由此得到最终行人类别的预测模型；步骤25、从实时采集的视频中筛选出所有的行人图像，并将这些行人图像全部送入到最终行人类别的预测模型中对其进行辨识分类，从中找出指定对象的所有行人图像。