[发明专利]对象识别网络的训练方法及系统

说明书

本公开涉及一种对象识别网络的训练方法及系统，该方法包括将训练集中的多个样本图像分别输入第一识别网络中进行处理，获取每个样本图像的多个第一视图的教师特征；将每个样本图像分别输入第二识别网络中进行处理，获取所述第二识别网络的第一网络损失，并根据所述教师特征获取所述第二识别网络的第二网络损失及第三网络损失；根据所述第一网络损失、第二网络损失及第三网络损失对所述第二识别网络进行训练，得到训练后的第二识别网络。通过上述方法训练得到的第二识别网络，可以实现对目标对象的精确识别，并且，本公开所述的训练方法具有扩展性好的有点，可以增加多个第一识别网络以对第二识别网络进行训练。

技术领域

本公开涉及机器学习技术领域，尤其涉及一种对象识别网络的训练方法及系统。

背景技术

行人重识别(英文Person Re-identification，简称ReID)可以实现用一张或多张行人图像在检索库中搜索该图像对应的人在其他视角相机中的图像。

早期的ReID技术采用人工设计的图像特征，精度很差，后来开始使用深度学习技术后精度有了大幅提高。当前主流的ReID技术为基于深度学习的ReID技术，通过深度学习实现ReID技术可以提高ReID技术的识别精度，然而，基于深度学习的ReID技术常常因为模型的训练结果不佳导致识别效率低下。

因此，急需提出一种新的训练方法，以提高基于深度学习的ReID技术的识别精度及工作效率。

发明内容

根据本公开的一方面，提供了一种对象识别网络的训练方法，所述方法包括：

将训练集中的多个样本图像分别输入第一识别网络中进行处理，获取每个样本图像的多个第一视图的教师特征；

将每个样本图像分别输入第二识别网络中进行处理，获取所述第二识别网络的第一网络损失，并根据所述教师特征获取所述第二识别网络的第二网络损失及第三网络损失；

根据所述第一网络损失、第二网络损失及第三网络损失对所述第二识别网络进行训练，得到训练后的第二识别网络，

其中，所述第二识别网络用于识别待处理图像中的目标对象的身份，所述第一识别网络为用于训练所述第二识别网络的教师网络。

在一种可能的实施方式中，所述第一识别网络包括第一视图分解子网络、第一图像增广子网络、第一卷积子网络、第一池化子网络以及第一嵌入子网络，

其中，将训练集中的多个样本图像分别输入第一识别网络中进行处理，获取每个样本图像的多个第一视图的教师特征，包括：

将目标样本图像输入第一视图分解子网络进行视图分解处理，获得所述目标样本图像的多个第一视图，其中，所述目标样本图像是所述多个样本图像中的任意一个；

将所述多个第一视图输入第一图像增广子网络进行增广处理，获得增广后的多个第二视图；

所述多个第一视图以及所述多个第二视图依次经由第一卷积子网络、第一池化子网络及第一嵌入子网络进行卷积、池化及嵌入处理，获得所述目标样本图像的多个第一视图的第一特征向量和多个第二视图的第二特征向量；

根据所述第一特征向量和所述第二特征向量，确定所述目标样本图像的教师特征。

在一种可能的实施方式中，所述第二识别网络包括特征提取网络、特征图映射网络及特征向量映射网络，所述特征提取网络包括第二图像增广子网络、第二卷积子网络、第二池化子网络、第二嵌入子网络以及分类子网络，

其中，将每个样本图像分别输入第二识别网络中进行处理，获取所述第二识别网络的第一网络损失，并根据所述教师特征获取所述第二识别网络的第二网络损失及第三网络损失，包括：