一种自监督预训练目标检测方法、系统、设备及存储介质

文档序号:33963340发布日期:2023-04-26 17:31阅读:来源:国知局

技术特征:

1.一种自监督预训练目标检测方法,其特征在于,包括以下步骤:

2.根据权利要求1所述的一种自监督预训练目标检测方法,其特征在于,所述步骤s1包括以下步骤:

3.根据权利要求1所述的一种自监督预训练目标检测方法,其特征在于,所述步骤s2还包括以下步骤:

4.根据权利要求1所述的一种自监督预训练目标检测方法,其特征在于,所述步骤s3包括以下步骤:

5.根据权利要求1所述的一种自监督预训练目标检测方法,其特征在于,所述步骤s3中,粘贴的patch块编码为object query的过程包括:

6.根据权利要求1所述的一种自监督预训练目标检测方法,其特征在于,所述步骤s4中,object query和合成图的特征输入至transformer中学习,并在transformer的解码器中自注意力模块引入为object query设计的注意力掩码。

7.利要求1所述的一种自监督预训练目标检测方法,其特征在于,所述步骤s4中,对学习后的object query进行类别和边界框预测的过程包括:

8.根据权利要求1所述的一种自监督预训练目标检测系统,其特征在于,包括patch块提取模块、合成图构成模块、多尺度特征转化模块和匹配模块;

9.一种终端设备,包括存储器、处理器以及存储在所述存储器中并可在所述处理器上运行的计算机程序,其特征在于,所述处理器执行所述计算机程序时实现如权利要求1-7任一项所述方法的步骤。

10.一种计算机可读存储介质,所述计算机可读存储介质存储有计算机程序,其特征在于,所述计算机程序被处理器执行时实现如权利要求1-7任一项所述方法的步骤。


技术总结
本发明公开了一种自监督预训练目标检测方法、系统、设备及存储介质,包括从给定的输入图片中提取proposals,选取前30个proposals作为带粘贴的patch块;获取的patch块粘贴到选取的背景图中,得到合成图,为预训练提供了准确的位置标注,提取下游待检测目标的色彩RGB值,在粘贴的patch块中随机选择一片区域将其改为与提取的色彩RGB值相应的颜色,优化了预训练目标检测中的分类能力,分别提取合成图的特征以及粘贴的patch块的多尺度特征,并将patch块的多尺度特征编码为object query;object query基于提取的合成图特征进行学习,并对学习后的object query进行类别和边界框预测,得到预测的集合,将预测的集合和真实的标注集合进行匹配,本发明解决了预训练代理任务在目标检测定位和分类方面的不足的问题。

技术研发人员:徐亦飞,李斌,严实,刘汕,王优,尉萍萍,宋毓韬,余乐
受保护的技术使用者:西安交通大学
技术研发日:
技术公布日:2024/1/11
当前第2页1 2 
网友询问留言 已有0条留言
  • 还没有人留言评论。精彩留言会获得点赞!
1