[发明专利]一种结合风格变换与姿态生成的行人重识别方法

权利要求书

1.一种结合风格变换与姿态生成的行人重识别方法，其特征在于，包括如下步骤：

步骤一、将图像原始数据集输入数据生成模块，获取不同风格的统一姿态图像；

步骤二、将不同风格的统一姿态图像输入分类识别模块，进行分类识别；

步骤三、输入查询图像，提取图像特征，与图像库中的图像特征进行相似性度量，进而完成行人重识别。

2.如权利要求1所述的一种结合风格变换与姿态生成的行人重识别方法，其特征在于：所述步骤一、将图像原始数据集输入数据生成模块，获取不同风格的统一姿态图像的具体过程是：

S201、输入图像原始数据集至风格转移模块，通过风格转移学习，输出具有不同相机风格的多张图像；

S202、输入图像原始数据集至姿态估计模块，通过AlphaPose进行姿态估计，提取出8种标准姿态骨架；

S203、将不同风格的图像与姿态骨架输入DCGAN网络，输出具有不同风格的统一姿态图像。

3.如权利要求1所述的一种结合风格变换与姿态生成的行人重识别方法，其特征在于：所述步骤二、将不同风格的统一姿态图像输入分类识别模块，进行分类识别的具体过程是：

S301、基于扩充的训练数据集训练分类识别网络，训练完成后，去掉网络模型的全连接层和分类器；

S302、将行人图库中所有图像作为网络输入，在全连接层之前进行输出，输出为行人特征。

4.如权利要求1所述的一种结合风格变换与姿态生成的行人重识别方法，其特征在于：所述步骤三、输入查询图像，提取图像特征，与图像库中的图像特征进行相似性度量，进而完成行人重识别的具体过程是：

S401、将查询图像作为分类识别网络输入，提取图像特征；

S402、使用欧式距离计算查询图像与图像库中的图像特征之间的相似度，根据相似度判断是否存在要查询图像中的行人，完成行人重识别任务。