CN110414432A 对象识别模型的训练方法、对象识别方法及相应的装置 （腾讯科技（深圳）有限公司）

本申请实施例提供了一种对象识别模型的以及分别与对象识别模型的输出连接的第一分的预测身份信息和样本图像的标注的身份信息第一训练损失值和第二训练损失值确定的总损2获取训练样本集，所述训练样本集中包括对象构建待训练模型，所述待训练模型包括所述对象识将各样本图像输入至所述待训练模型中，根据所述第一根据所述总损失值对所述待训练模型的模型参数进行根据所述第二分类模型的模型参数，确定所述训练样本集计算所述第二分类模型的模型参数中，同一对象的7.根据权利要求1至6中任一项所述方法，将所述样本图像输入至预训练好的朝向分类模型中，基3第一损失确定模块，用于将各样本图像输入至模型输出的预测身份信息和样本图像的身份标签所对应的身份信息确定第一训练损失值，第二损失确定模块，用于基于所述第一训练损失值和所述参数调整模块，用于根据所述总损失值对所述待训特征提取模块，用于将所述待识别的对象图像输入对4[0010]根据总损失值对待训练模型的模型参数进行调整，直至总损5模型是通过第一方面或第一方面的任一可选实施方式中所提供的对象识别模型的训练方6[0050]本申请上述的和/或附加的方面和优点从下面结合附图对实施例的描述中将变得7[0063]人工智能(ArtificialIntelligence，AI)是利用数字计算机或者数字计算机控[0065]计算机视觉技术(ComputerVision，CV)计算机视觉是一门研究如何使机器“看”8中行人朝向的多样性是制约图像识别准确率的一大问题。但是在现有的图像识别方法中，是用于提取图像特征的特征提取网络，如骨干网络(BackboneNetwork)、残差网络9身份标签和朝向标签所对应的身份信息及朝向信息的差异来优化待训练模型的模型参数。选的，第一损失函数或第二损失函数可以为crossentropy(交叉熵)损失、lsr(Label施例还基于第一损失函数和第二损失函数为待训练模型设置了总损失函数，在训练过程第二训练损失值，实现了基于图像中对象的身份信息和对象的朝向信息对模型的优化训而使得待训练模型(包括对象识别模型)在训练过程中学习到相同身份不同朝向的对象的样本图像之间的相似性，即同一对象的不同朝向的图像的预测结果之间的差异也尽可能所有元素即为一个神经元的网络参数。由于第二分类模型的每个神经元对应一个分类结块的神经元的数量则为C*M种，假设对象识别模型的输出为N维列向量(如对象识别模型用元的预测结果在目标空间的参数向量。获取每一相同身份不同朝向的预测结果的参数向[0106]假设该权重矩阵Q中第一行对应于第二分类模块中预测结果为第一身份第一朝向参数向量为(a11,a12,a13,a14)，第一身份第二朝向不同朝向的数量，Dk和Dl分别表示n个朝向中的第身份的对象的前向样本图像的预测结果与背向样本图像的预测结果两者在目标空间中的[0113]表示第j个身份的对象的背向样本图像的预测结果与侧向样本图一个或多个损失函数的权重，例如，上述第三损失函数中，第三损失函数还可以表示为[0129]其中，loss为总损失函数，lossid为第一损失函数(其值即为第一训练损失值)，[0131]具体的，在训练过程中将第一训练损失值和第三训练损失值代入上述总损失函类模型包括依次级联的卷积神经网络(还可以是骨干网络或其他特征提取网络等)和全连种角度关系，则可以将主体与参考基准之间角度在特定范围内的对象的朝向定义为正向，[0142]当然，如果采用人工标注方式或者模型预测的方式来得到样本图像的朝向标签[0148]下面结合图3b对上述各损失函数的意义进行进一步说明，图3b中的每个小黑点、view约束该ID1行人的不同背向的图像之间的预测结果尽可能相同(图中箭头所示的类模块以及第二分类模块组成的待训练模型整体进行训练，在总损失值满足预设条件后，[0158]本申请实施例提供的一种对象识别方法，利用对象识别模型对对象图像进行识应于朝向信息设置的训练损失值，使得训练好的对象识别模型对于对象朝向的鲁棒性更同监控视野的监控设备中获取的图像或视频序列称为gallery图像(参考图像)。将该对象的图像特征和gallery图像的图像特征，并获取query图像的图像特征和每一gallery图像[0168]图5为本申请实施例提供的一种对象识别模型的训练装置的结[0172]第二损失确定模块54用于基于第一训练损失值和第二训练损失值，确定总损失合构成待训练模型后，再基于带有身份标签和朝向标签的样本图像对待训练模型进行训处理器及存储在存储器上并可在处理器上运行的计算机程序，处理器执行该计算机程序中对象的预测身份信息，第二分类模型用于输出样本图像中对象的预测身份及朝向信息；[0203]可以理解的是，介质中存储的可以是对象识别模型的训练方法对应的计算机程[0208]存储器703可以是ROM或可存储静态信息和指令的其他类型的静态存储设备，RAM[020