一种目标检测模型的训练方法、目标检测方法及相关装置与流程

技术特征：
1.一种目标检测模型的训练方法，其特征在于，所述的训练方法包括：对训练图片的标注信息执行预设次数的添加噪声操作，以得到预设数量的噪声标注信息，其中，所述标注信息包括目标框和所述目标框对应的目标类别；根据噪声标注信息生成的去噪组，以得到预设数量的去噪组；基于预设数量的去噪组、初始化生成的匹配组以及所述训练图片，对目标检测模型进行训练。2.根据权利要求1所述的目标检测模型的训练方法，其特征在于，所述预设数量的噪声标注信息中的各噪声标注信息互不相同。3.根据权利要求1所述的目标检测模型的训练方法，其特征在于，所述添加噪声操作包括对目标框的添加框噪声操作和/或对目标类别添加类别噪声操作。4.根据权利要求3所述的目标检测模型的训练方法，其特征在于，所述标注信息包括多个目标框，所述添加噪声操作为分别对多个目标框中的每个目标框执行添加框噪声操作，以及分别对每个目标框对应的目标类别执行添加类别噪声操作。5.根据权利要求3所述的目标检测模型的训练方法，其特征在于，所述目标框包括目标框中心点坐标、目标框宽度和目标框高度；所述对目标框添加框噪声操作具体包括：为所述目标框随机生成目标框噪声，其中，目标框噪声包括目标框中心点坐标噪声、目标框宽度噪声和目标框高度噪声中的一种或者多种；基于所述目标框噪声对所述目标框进行调整，以得到噪声目标框。6.根据权利要求5所述的目标检测模型的训练方法，其特征在于，所述目标框中心点坐标噪声中的x轴坐标的绝对值小于，目标框中心点坐标噪声中的y轴坐标的绝对值小于；目标框宽度噪声的绝对值小于，目标框高度噪声的绝对值小于，其中，和均为0-1间的数值，表示目标框宽度，表示目标框高度。7.根据权利要求3所述的目标检测模型的训练方法，其特征在于，所述对目标类别添加类别噪声操作具体包括：按照预设概率将所述目标类别替换为噪声类别，其中，所述噪声类别包含于所述训练图片所属的训练样本集对应的类别集合内，并且所述噪声类别所述目标类别不同。8.根据权利要求1所述的目标检测模型的训练方法，其特征在于，所述根据噪声标注信息生成去噪组，以得到预设数量的去噪组具体包括：将所述噪声标注信息中的噪声目标框转换为噪声目标框向量，并将所述噪声目标框对应的噪声类别转换为噪声类别向量；将所述噪声目标框向量与所述噪声类别向量连接生成去噪query，根据生成的去噪query形成去噪组。9.根据权利要求1-8任意一项所述的目标检测模型的训练方法，其特征在于，所述基于预设数量的去噪组、初始化生成的匹配组以及所述训练图片，对目标检测模型进行训练具体包括：基于所述训练图片及所述目标检测模型的编码模块生成若干特征图块；基于所述若干特征图块、预设数量的去噪组、初始化生成的匹配组及所述目标检测模
型的解码模块，生成预测信息；基于所述预测信息和所述标注信息对目标检测模型的训练进行调整。10.根据权利要求9所述的目标检测模型的训练方法，其特征在于，所述预测信息包括各去噪组各自对应的第一预测信息和所述匹配组对应的第二预测信息，其中，所述第二预测信息为通过匈牙利匹配所得到的。11.根据权利要求10所述的目标检测模型的训练方法，其特征在于，所述基于所述若干特征图块、预设数量的去噪组、初始化生成的匹配组及所述目标检测模型的解码模块，生成预测信息具体包括：基于预设数量的去噪组和初始化生成的匹配组确定注意力掩码矩阵；将若干特征图块、预设数量的去噪组、初始化生成的匹配组以及注意力掩码矩阵输入所述目标检测模型的解码模块，通过所述解码模块输出预测信息。12.根据权利要求11所述的目标检测模型的训练方法，其特征在于，所述目标检测模型的解码模块配置有注意力机制，其中，所述若干特征图块为注意力机制中的key值和value值；所述去噪组、初始化生成的匹配组以及注意力掩码矩阵为注意力机制中的query值。13.根据权利要求11所述的目标检测模型的训练方法，其特征在于，所述注意力掩码矩阵的行数和列数均等于各去噪组包括的去噪query的数量和匹配组中的query的数量之和，并且各去噪组中的去噪query的向量维度均等于匹配组中的匹配query的向量维度。14.根据权利要求13所述的目标检测模型的训练方法，其特征在于，所述基于预设数量的去噪组和初始化生成的匹配组确定注意力掩码矩阵具体包括：将各去噪组所包含的query和匹配组所包含的query按照行列排列，以形成初始矩阵；对于初始矩阵中的每个矩阵元素，若对应的query和对应的query属于不相同去噪组，或者，对应的query属于匹配组对应的query属于匹配组，则将的矩阵元素值设置为第一数值；若对应的query和对应的query属于相同去噪组，或者，对应的query属于匹配组，则将的矩阵元素值设置为第二数值，以得到注意力掩码矩阵，其中，第一数值用于行的query能得到第列的query所携带的信息，第二数值用于表示第行的query能得到第列的query所携带的信息。15.一种目标检测方法，其特征在于，所述方法应用采用如权利要求1-14任意一项所述的目标检测模型的训练方法训练得到目标检测模型，所述方法包括：将待检测图片输入所述目标检测模型；通过所述目标检测模型输出所述待检测图片的检测框及检测类别。16.一种目标检测模型的训练装置，其特征在于，所述的训练装置包括：添加模块，用于对训练图片的标注信息执行预设次数的添加噪声操作，以得到预设数量的噪声标注信息，其中，所述标注信息包括目标框和所述目标框对应的目标类别；生成模块，用于根据噪声标注信息生成去噪组，以得到预设数量的去噪组；训练模块，用于基于预设数量的去噪组、初始化生成的匹配组以及所述训练图片，对目标检测模型进行训练。17.一种计算机可读存储介质，其特征在于，所述计算机可读存储介质存储有一个或者多个程序，所述一个或者多个程序可被一个或者多个处理器执行，以实现如权利要求1-14
任意一项所述的目标检测模型的训练方法中的步骤，和/或以实现如权利要求15所述的目标检测方法中的步骤。18.一种终端设备，其特征在于，包括：处理器、存储器及通信总线;所述存储器上存储有可被所述处理器执行的计算机可读程序；所述通信总线实现处理器和存储器之间的连接通信；所述处理器执行所述计算机可读程序时实现如权利要求1-14任意一项所述的目标检测模型的训练方法中的步骤，和/或实现如权利要求15所述的目标检测方法中的步骤。

技术总结
本申请公开了一种目标检测模型的训练方法、目标检测方法及相关装置，所述训练方法包括对训练图片的标注信息执行预设次数的添加噪声操作，以得到预设数量的噪声标注信息；根据噪声标注信息生成的去噪组，以得到预设数量的去噪组；基于预设数量的去噪组、初始化生成的匹配组以及所述训练图片，对目标检测模型进行训练。本申请通过对标注信息添加噪声来增加形成噪声标注信息，并将基于噪声标注信息生成去噪组作为目标检测模型的输入项，以增加目标检测模型对标注信息的学习，从而可以提高目标检测模型对标注信息的学习速度，进而提高目标检测模型的训练速度。检测模型的训练速度。检测模型的训练速度。


技术研发人员：李峰 张浩 刘世隆 张磊
受保护的技术使用者：粤港澳大湾区数字经济研究院（福田）
技术研发日：2022.02.28
技术公布日：2022/4/1