基于知识蒸馏的行人重识别方法、装置、设备及存储介质

摘要

本发明提供一种基于知识蒸馏的行人重识别方法、装置、设备及存储介质，方法包括：基于训练好的行人重识别模型对待识别的图像数据进行行人重识别，得到行人重识别结果；以及还包括：基于数据训练集和所述数据训练集所对应的预设教师模型集，得到第一输出数据；基于所述数据训练集对神经网络进行训练，得到学生模型以及所述学生模型输出的第二输出数据；基于所述第一输出数据、所述第二输出数据和判别器网络计算所述学生模型的损失函数；调整所述学生模型的模型参数以使所述损失函数最小化，得到所述训练好的行人重识别模型。根据本发明，基于跨域数据集和知识蒸馏的学习方法训练行人重识别模型，从而提高行人重识别的精确度。