目标检测模型的训练方法、目标检测方法、装置和介质.pdf

(19)国家知识产权局(12)发明专利申请(10)申请公布号CN115082758A(43)申请公布日2022.09.20(21)申请号202210999180.3(22)申请日2022.08.19(71)申请人深圳比特微电子科技有限公司地址518051广东省深圳市南山区粤海街道高新南六道航盛科技大厦801(72)发明人涂鹏凌明杨作兴杨敏艾国(74)专利代理机构北京润泽恒知识产权代理有限公司11319专利代理师刘祥景(51)Int.Cl.G06V10/774(2022.01)G06V10/80(2022.01)权利要求书3页说明书17页附图3页(54)发明名称目标检测模型的训练方法、目标检测方法、装置和介质(57)摘要本申请实施例提供了一种目标检测模型的训练方法、目标检测方法、装置和介质，其中的训练方法具体包括：利用特征提取单元，对图像样本进行特征提取；利用特征融合单元，对图像特征进行融合处理；分别对融合图像特征进行第一随机掩码处理和第二随机掩码处理，以得到第一掩码融合图像特征和第二掩码融合图像特征；利用第一检测单元，对第一掩码融合图像特征进行目标检测，得到第一检测结果，利用第二检测单元，对第二掩码融合图像特征进行目标检测，得到第二检测结果；根据第一检测结果和第二检测结果之间的匹配信息，确定损失信息，并根据损失信息，对目标检测模型的参数进行更新。本申请实施例可以使目标检测模型在环境发生变化的情况下更加鲁棒。CN115082758ACN115082758A权利要求书1/3页1.一种目标检测模型的训练方法，其特征在于，所述目标检测模型包括：特征提取单元、特征融合单元和检测单元，所述检测单元包括：第一检测单元和第二检测单元，所述方法包括：利用特征提取单元，对图像样本进行特征提取，以得到图像特征；利用特征融合单元，对所述图像特征进行融合处理，以得到融合图像特征；分别对所述融合图像特征进行第一随机掩码处理和第二随机掩码处理，以得到第一掩码融合图像特征和第二掩码融合图像特征；利用第一检测单元，对所述第一掩码融合图像特征进行目标检测，以得到第一检测结果，以及，利用第二检测单元，对所述第二掩码融合图像特征进行目标检测，以得到第二检测结果；根据所述第一检测结果和第二检测结果之间的匹配信息，确定损失信息，并根据所述损失信息，对所述目标检测模型的参数进行更新。2.根据权利要求1所述的方法，其特征在于，所述第一随机掩码处理、所述第二随机掩码处理，包括：随机确定概率值；在概率值大于概率阈值的情况下，对所述融合图像特征执行随机掩码操作。3.根据权利要求2所述的方法，其特征在于，所述对所述融合图像特征执行随机掩码操作，包括：根据所述图像样本对应的图像尺寸，随机确定位置点；根据所述位置点和掩码尺寸，确定掩码区域；将所述掩码区域中融合图像特征的特征值替换为预设特征值。4.根据权利要求1所述的方法，其特征在于，所述第一检测结果或第二检测结果对应的检测结果信息包括如下概率信息中的至少一种：所述图像样本中像素点属于预设目标的像素概率信息；所述图像样本中位置区域属于预设目标的区域概率信息；以及所述图像样本中像素点属于图像前景的前景概率信息。5.根据权利要求4所述的方法，其特征在于，所述第一检测结果包括：第一概率信息，所述第二检测结果包括：第二概率信息；所述根据所述第一检测结果和第二检测结果之间的匹配信息，确定损失信息，包括：确定第一概率信息与第二概率信息之间的度量信息；根据所述度量信息，确定损失信息。6.根据权利要求5所述的方法，其特征在于，所述度量信息为对称性度量信息；所述确定第一概率信息与第二概率信息之间的度量信息，包括：确定第一概率信息的对数与第二概率信息的对数的第一差值，并确定第一概率信息的对数与第一差值之间的第一乘积；确定第二概率信息的对数与第一概率信息的对数的第二差值，并确定第二概率信息的对数与第二差值之间的第二乘积；根据所述第一乘积和第二乘积的和，确定第一概率信息与第二概率信息之间的对称性度量信息。2CN115082758A权利要求书2/3页7.根据权利要求5所述的方法，其特征在于，所述根据所述度量信息，确定损失信息，包括：确定第一概率信息与所述图像样本的标注信息之间的第一损失信息；确定第二概率信息与所述图像样本的标注信息之间的第二损失信息；根据所述度量信息、所述第一损失信息和所述第二损失信息，确定损失信息。8.一种目标检测方法，其特征在于，所述方法包括：接收待检测图像；利用目标检测模型，对所述待检测图像进行目标检测，以得到对应的检测结果；其中，所述目标检测模型包括：特征提取单元、特征融合单元和检测单元，所述检测单元包括：第一检测单元和第二检测单元；所述目标检测模型的训练过程包括：利用特征提取单元，对图像样本进行特征提取，以得到图像特征；利用特征融合单元