[发明专利]行人重识别模型的训练方法、装置、计算设备及存储介质

权利要求书

1.一种行人重识别模型的训练方法，其特征在于，包括：

获取第一行人重识别训练数据集，所述第一行人重识别训练数据集包括至少两个不同的训练图像集；

将所述第一行人重识别训练数据集输入预设的行人重识别模型进行处理，并基于处理结果计算第一损失函数；

对所述第一行人重识别训练数据集进行处理得到第二行人重识别训练数据集；

将所述第二行人重识别训练数据集输入监督网络进行处理，并根据处理结果计算第二损失函数，所述第二损失函数包括聚类损失函数和重构损失函数，所述监督网络包括：

预分类模块，用于使用预训练的分类模型从输入的所述第二行人重识别训练数据集提取图像特征；

降维模块，用于对所述分类模提取的图像特征进行降维处理；

聚类模块，用于对降维后的图像特征进行聚类处理，以使属于同一行人的图像特征归到一类，根据聚类结果计算所述聚类损失函数，并根据所述聚类损失函数的计算结果调整所述聚类模块；

重构模块，用于利用反卷积网络根据所述聚类模块的聚类中心进行重构，得到重构图像，根据重构结构计算所述重构损失函数，并根据所述重构损失函数的计算结果调整所述重构模块；

根据所述第一损失函数和第二损失函数的计算结果调整所述预设的行人重识别模型，得到目标行人重识别模型。

2.根据权利要求1所述的方法，其特征在于，所述聚类损失函数为：

L_rim＝R(θ)-λ[H(Y)-H(Y|X)],

其中，R(θ)是正则项，λ是权重因子，H(Y)和H(Y|X)分别代表熵和条件熵，

y＝softmax_θ(x)

x为输入的特征向量。

3.根据权利要求1所述的方法，其特征在于，所述重构损失函数为：

其中，X_j代表输入图像中属于类别j的集合，r_j是类别j的聚类中心的向量表示，D代表所使用的反卷积神经网络；

G代表所使用的预训练分类模型。

4.一种行人重识别方法，其特征在于，包括：

获取待识别的行人图像；

利用权利要求1-3中的任一项所述的训练方法训练的目标行人重识别模型进行行人重识别。

5.一种行人重识别模型的训练装置，其特征在于，包括：

第一数据获取模块，用于获取第一行人重识别训练数据集，所述第一行人重识别训练数据集包括至少两个不同的训练图像集；

第一训练模块，用于将所述第一行人重识别训练数据集输入预设的行人重识别模型进行处理，并基于处理结果计算第一损失函数；

第二数据获取模块，用于对所述第一行人重识别训练数据集进行处理得到第二行人重识别训练数据集；

第二训练模块，用于将所述第二行人重识别训练数据集输入监督网络进行处理，并根据处理结果计算第二损失函数，所述第二损失函数包括聚类损失函数和重构损失函数，所述监督网络包括：

预分类模块，用于使用预训练的分类模型从输入的所述第二行人重识别训练数据集提取图像特征；

降维模块，用于对所述分类模提取的图像特征进行降维处理；

聚类模块，用于对降维后的图像特征进行聚类处理，以使属于同一行人的图像特征归到一类，根据聚类结果计算所述聚类损失函数，并根据所述聚类损失函数的计算结果调整所述聚类模块；

重构模块，用于利用反卷积网络根据所述聚类模块的聚类中心进行重构，得到重构图像，根据重构结构计算所述重构损失函数，并根据所述重构损失函数的计算结果调整所述重构模块；

调整模块，用于根据所述第一损失函数和第二损失函数的计算结果调整所述预设的行人重识别模型，得到目标行人重识别模型。

6.一种行人重识别装置，其特征在于，包括：

获取模块，用于获取待识别的行人图像；

识别模块，用于利用权利要求1-3中的任一项所述的训练方法训练的目标行人重识别模型进行行人重识别。