本申请公开了一种目标重识别方法、模型训练方法、装置、设备及存储介质，属于图像处理领域。所述方法包括：获取第一目标框图像，第一目标框图像是对待检测图像进行目标检测得到；通过空间变换网络模型对第一目标框图像进行校正，得到第二目标框图像；通过重识别模型对第二目标框图像进行重识别，得到目标特征。如此，可以先通过空间变换网络模型将检测的非标准目标框图像校正为标准目标框图像，再作为重识别模型的输入图像进行重识别，这样可以保证重识别模型的输入图像与其训练过程中使用的标准目标框图像保持一致，避免重识别模型的输入图像与其训练过程中使用的标准目标框图像不一致对重识别模型精度的影响，提高重识别结果的准确度。的准确度。的准确度。
【技术实现步骤摘要】
目标重识别方法、模型训练方法、装置、设备及存储介质
[0001]本申请涉及图像处理领域，特别涉及一种目标重识别方法、模型训练方法、装置、设备及存储介质。
技术介绍
[0002]目标ReID(Re
‑
Identification，重识别)是指利用图像处理技术来识别图像中的特定目标，其在智能视频监控、机器人以及自动驾驶等领域发挥着必不可少的作用。目标ReID主要包括行人ReID或车辆ReID等。比如，在行人ReID中，可以利用图像处理技术来判定某一摄像头下的行人是否出现在其他摄像头中，以描绘行人的活动路径，实现跨镜头追踪等目的。
[0003]相关技术中，可以利用重识别模型进行重识别，重识别模型的输入为目标框图像，输出为目标框图像中目标的目标特征，目标特征用于确定目标的类别。在进行重识别之前，需要预先基于训练数据训练得到该重识别模型，该训练数据包括多个标准目标框图像以及每个标准目标框对应的目标类别。其中，该训练数据可以通过人工标注得到，标准目标框图像中包括完整的目标，且目标与目标框的位置匹配。在训练得到重识别模型之后进行重识别时，可以先通过检测器对待检测图像进行目标检测，得到目标框图像，再通过重识别模型对目标框图像进行重识别，得到目标特征。
[0004]但是，检测器输出的目标框图像与训练过程中所使用的标准目标框图像可能不一致，比如检测器输出的目标框图像中存在不完整的目标或者存在目标框漂移等情况，这些情况将会影响重识别模型的精度，进而影响重识别结果的准确度。
技术实现思路
[0005]本申请实施例提供了一种目标重识别方法、模型训练方法、装置、设备及存储介质，可以用于解决相关技术中存在的重识别结果准确度较低的问题。所述技术方案如下：
[0006]一方面，提供了一种目标重识别方法，所述方法包括：
[0007]获取第一目标框图像，所述第一目标框图像是对待检测图像进行目标检测得到；
[0008]通过空间变换网络模型对所述第一目标框图像进行校正，得到第二目标框图像；
[0009]通过重识别模型对所述第二目标框图像进行重识别，得到目标特征。
[0010]可选地，所述空间变换网络模型包括参数预测网络、坐标映射网络和像素采集网络；所述通过空间变换网络模型对所述第一目标框图像进行校正，得到第二目标框图像，包括：
[0011]将所述第一目标框图像作为所述参数预测网络的输入，通过所述参数预测网络对所述第一目标框图像进行参数预测，得到空间变换参数；
[0012]基于所述空间变换参数，通过所述坐标映射网络，构建所述第一目标框图像与校正后的第一目标框图像之间的空间映射关系；
[0013]基于所述空间映射关系，通过所述像素采集网络，对所述第一目标框图像进行像
素采集，得到所述第二目标框图像。
[0014]可选地，所述通过所述参数预测网络对所述第一目标框图像进行参数预测，得到空间变换参数，包括：
[0015]通过所述空间变换网络模型对所述第一目标框图像进行参数预测，得到初始变换矩阵；
[0016]对所述初始变换矩阵进行矩阵变换，将矩阵变换后的初始变换矩阵作为所述空间变换参数。
[0017]可选地，所述参数预测网络包括卷积层、池化层和全连接层。
[0018]另一方面，提供了一种目标重识别装置，所述装置包括：
[0019]获取模块，用于获取第一目标框图像，所述第一目标框图像是对待检测图像进行目标检测得到；
[0020]校正模块，用于通过空间变换网络模型对所述第一目标框图像进行校正，得到第二目标框图像；
[0021]重识别模块，用于通过重识别模型对所述第二目标框图像进行重识别，得到目标特征。
[0022]另一方面，提供了一种模型训练方法，所述方法包括：
[0023]获取多个第一样本图像，所述多个第一样本图像属于标准目标框图像；
[0024]对所述多个第一样本图像进行处理，得到所述多个第一样本图像一一对应的多个第二样本图像，所述多个第二样本图像包括至少一个非标准目标框图像；
[0025]通过所述重识别模型对所述多个第一样本图像进行重识别，得到所述多个第一样本图像的重识别结果；
[0026]通过待训练空间变换网络模型对所述多个第二样本图像进行校正，通过所述重识别模型对校正后的多个第二样本图像进行重识别，得到所述多个第二样本图像的重识别结果；
[0027]基于所述多个第一样本图像的重识别结果与所述多个第二样本图像的重识别结果之间的重识别误差，对所述待训练空间变换网络模型的模型参数进行更新，将模型参数更新后的待训练空间变换网络模型作为训练好的空间变换网络模型。
[0028]另一方面，提供了一种模型训练装置，所述装置包括：
[0029]第一获取模块，用于获取多个第一样本图像，所述多个第一样本图像属于标准目标框图像；
[0030]图像处理模块，用于对所述多个第一样本图像进行处理，得到所述多个第一样本图像一一对应的多个第二样本图像，所述多个第二样本图像包括至少一个非标准目标框图像；
[0031]第一重识别模块，用于通过所述重识别模型对所述多个第一样本图像进行重识别，得到所述多个第一样本图像的重识别结果；
[0032]第二重识别模型，用于通过待训练空间变换网络模型对所述多个第二样本图像进行校正，通过所述重识别模型对校正后的多个第二样本图像进行重识别，得到所述多个第二样本图像的重识别结果；
[0033]更新模块，用于基于所述多个第一样本图像的重识别结果与所述多个第二样本图
像的重识别结果之间的重识别误差，对所述待训练空间变换网络模型的模型参数进行更新，将模型参数更新后的待训练空间变换网络模型作为训练好的空间变换网络模型。
[0034]另一方面，提供了一种计算机设备，所述计算机设备包括：
[0035]处理器；
[0036]用于存储处理器可执行指令的存储器；
[0037]其中，所述处理器被配置为执行上述目标重识别方法。
[0038]另一方面，提供了一种计算机可读存储介质，所述计算机可读存储介质上存储有指令，所述指令被处理器执行时实现上述目标重识别方法或模型训练方法。
[0039]另一方面，还提供了一种计算机程序产品，当所述计算机程序产品被执行时，用于实现上述目标重识别方法或模型训练方法。
[0040]本申请实施例提供的技术方案带来的有益效果是：
[0041]本申请实施例中，对于对待检测图像进行目标检测得到的第一目标框图像，可以先通过空间变换网络模型对第一目标框图像进行校正，得到第二目标框图像，再通过重识别模型对第二目标框图像进行重识别。如此，可以通过空间变换网络模型，将检测的非标准目标框图像校正为标准目标框图像，再将标准目标框图像作为重识别模型的输入图像进行重识别，这样可以保证重识别模型的输入图像与其训练过程中使用的标准目标框图像保持一致，从而避免重识别模型的输入图像本文档来自技高网...
【技术保护点】
【技术特征摘要】
1.一种目标重识别方法，其特征在于，所述方法包括：获取第一目标框图像，所述第一目标框图像是对待检测图像进行目标检测得到；通过空间变换网络模型对所述第一目标框图像进行校正，得到第二目标框图像；通过重识别模型对所述第二目标框图像进行重识别，得到目标特征。2.如权利要求1所述的方法，其特征在于，所述通过空间变换网络模型对所述第一目标框图像进行校正，得到第二目标框图像，包括：将所述第一目标框图像作为所述空间变换网络模型的输入，通过所述空间变换网络模型对所述第一目标框图像进行参数预测，得到空间变换参数；通过所述空间变换网络模型，基于所述空间变换参数对所述第一目标框图像进行空间变换，得到所述第二目标框图像。3.如权利要求2所述的方法，其特征在于，所述基于所述空间变换参数对所述第一目标框图像进行空间变换，得到所述第二目标框图像，包括：基于所述空间变换参数，构建所述第一目标框图像与所述第二目标框图像之间的空间映射关系；基于所述空间映射关系，对所述第一目标框图像进行像素采集，得到所述第二目标框图像。4.一种模型训练方法，其特征在于，所述方法包括：获取多个第一样本图像，所述多个第一样本图像属于标准目标框图像；对所述多个第一样本图像进行处理，得到所述多个第一样本图像一一对应的多个第二样本图像，所述多个第二样本图像包括至少一个非标准目标框图像；通过所述重识别模型对所述多个第一样本图像进行重识别，得到所述多个第一样本图像的重识别结果；通过待训练空间变换网络模型对所述多个第二样本图像进行校正，通过所述重识别模型对校正后的多个第二样本图像进行重识别，得到所述多个第二样本图像的重识别结果；基于所述多个第一样本图像的重识别结果与所述多个第二样本图像的重识别结果之间的重识别误差，对所述待训练空间变换网络模型的模型参数进行更新，将模型参数更新后的待训练空间变换网络模型作为训练好的空间变换网络模型。5.如权利要求4所述的方法，其特征在于，所述对所述多个第一样本图像分别进行处理，包括：对于所述多个第一样本图像中的第一图像，对所述第一图像进行目标截断，基于目标截断后的第一图像确定所述第一图像对应的第二样本图像，所述第一图像对应的第二样本图像包括不完整的目标，所述第一图像为所述多个第一样本图像中的任一个；或者，对所述第一图像进行目标框漂移处理，基于目标框漂移处理后的第一图像，确定所述第一图像对应的第二样本图像，所述第一图像...
【专利技术属性】
技术研发人员：李帅杰，魏新明，
申请(专利权)人：深圳云天励飞技术股份有限公司，
类型：发明
国别省市：
还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。