神经网络、目标检测方法、神经网络训练方法及相关产品
摘要:
本申请实施例公开了一种用于目标检测的神经网络、目标检测方法、神经网络训练方法及相关产品,该用于目标检测的神经网络包括编码网络和解码网络,解码网络包括第一解码层;编码网络,用于对待检测图像进行特征提取,得到多个第一特征图;第一解码层,用于对目标第一特征图和多个目标参考向量进行注意力处理,得到多个第一参考向量;对多个第一参考向量进行目标检测,得到多个第一候选框和多个第一类别;对多个第一候选框中的每个所述第一候选框对应的特征图和多个第一参考向量中的每个第一参考向量进行处理,得到多个第二参考向量;基于多个第二参考向量和多个第一候选框,得到多个第二候选框。本申请实施例有利于提高目标检测精度。
0/0