摘要
本申请实施例提供了一种目标检测方法、装置、计算机设备和存储介质,所述目标检测方法包括:对预设视觉自注意力模型、预设编码器、预设解码器和预设一对一集合匹配模型进行联合训练,得到训练好的视觉自注意力模型、编码器、解码器和一对一集合匹配模型;将待检测图像、视觉自注意力模型输出的第一图像特征、编码器输出的第二图像特征和解码器输出的第三图像特征分别输入训练好的视觉自注意力模型、编码器、解码器和一对一集合匹配模型中,得到待检测图像中的目标对象,其中,在视觉自注意力模型中生成随机掩码,将随机掩码与注意力计算融合,提升注意力机制的鲁棒性,减少对冗余特征的依赖,提高模型的抗噪声能力。
技术关键词
注意力模型
多尺度特征提取
解码器
编码器
视觉
图像
矩阵
计算机设备
参数
对象
通道
采样点
冗余特征
生成随机
注意力机制
表达式
处理器
输入模块