一种目标检测方法、装置及电子设备与流程

技术特征：
1.一种目标检测方法，其特征在于，所述方法应用于电子设备，所述电子设备预存有训练完成的目标检测模型，所述目标检测模型包括编码器和解码器，所述方法包括：根据所述待检测图像的特征张量确定第一查询特征、第一键特征和第一值特征；通过所述编码器对所述第一查询特征、所述第一键特征和所述第一值特征进行空间位置自注意力处理，得到空间位置自注意力结果；根据所述空间位置自注意力结果，确定所述编码器输出的编码数据；其中，所述空间位置自注意力处理为将所述第一键特征进行行列解耦，并根据行列解耦得到的第一键行特征、第一键列特征以及所述第一查询特征和所述第一值特征，确定所述空间位置自注意力结果；通过所述解码器对第二查询特征、第二键特征、第二值特征以及所述编码数据进行处理，得到解码数据，其中，所述第二查询特征、所述第二键特征和所述第二值特征通过预设参数确定；根据所述解码数据，确定所述待检测图像对应的检测结果，所述检测结果包括所述待检测图像中的目标对象对应的目标特征集。2.根据权利要求1所述的方法，其特征在于，根据行列解耦得到的第一键行特征、第一键列特征以及所述第一查询特征和所述第一值特征，确定所述空间位置自注意力结果的步骤，包括：根据所述第一键行特征和所述第一查询特征，确定行注意力图；根据所述第一键列特征和所述第一查询特征，确定列注意力图；根据所述行注意力图、所述列注意力图和所述第一值特征，确定所述空间位置自注意力结果。3.根据权利要求2所述的方法，其特征在于，根据所述行注意力图、所述列注意力图和所述第一值特征，确定所述空间位置自注意力结果的步骤，包括：根据所述行注意力图和所述列注意力图，对所述第一值特征进行加权求和处理，得到所述空间位置自注意力结果。4.根据权利要求1所述的方法，其特征在于，所述编码器还包括层级自注意力模块，根据所述空间位置自注意力结果，确定所述编码器输出的编码数据的步骤，包括：通过所述层级自注意力模块对所述空间位置自注意力结果进行处理，得到层级自注意力结果；根据所述层级自注意力结果，确定编码数据。5.根据权利要求4所述的方法，其特征在于，根据所述层级自注意力结果，确定编码数据的步骤，包括：判断所述层级自注意力结果是否满足预设的编码器输出条件；如果否，将所述层级自注意力结果作为所述待检测图像的特征张量，继续执行所述根据所述待检测图像的特征张量确定第一查询特征、第一键特征和第一值特征；通过所述编码器对所述第一查询特征、所述第一键特征和所述第一值特征进行空间位置自注意力处理，得到空间位置自注意力结果的步骤，以及通过所述层级自注意力模块对所述空间位置自注意力结果进行处理，得到层级自注意力结果的步骤，直到所述层级自注意力结果满足所述编码器输出条件，将所述层级自注意力结果作为编码数据。6.根据权利要求1所述的方法，其特征在于，所述预设参数包括预设的初始位置锚点和
预设的初始模式编码，所述第二查询特征、所述第二键特征和所述第二值特征通过以下步骤确定：根据所述初始位置锚点和预设的位置编码函数，确定位置编码；根据所述位置编码和所述初始模式编码，确定所述第二查询特征和所述第二键特征；根据所述初始模式编码，确定所述第二值特征。7.根据权利要求6所述的方法，其特征在于，根据所述位置编码和所述初始模式编码，确定所述第二查询特征和所述第二键特征的步骤，包括：将所述位置编码与所述初始模式编码组合，得到第二查询特征和第二键特征；根据所述初始模式编码，确定所述第二值特征的步骤，包括：将所述初始模式编码确定为所述第二值特征。8.根据权利要求7所述的方法，其特征在于，所述初始模式编码为初始模式编码集，所述初始模式编码集中至少包括2个不同的模式编码。9.根据权利要求1
‑
8任一项所述的方法，其特征在于，通过所述解码器对第二查询特征、第二键特征、第二值特征以及所述编码数据进行处理，得到解码数据的步骤，包括：通过所述解码器对所述第二查询特征、所述第二键特征和所述第二值特征进行自注意力处理，得到自注意力结果；通过所述解码器对所述自注意力结果以及所述编码数据进行空间位置注意力处理，得到解码数据。10.根据权利要求9所述的方法，其特征在于，通过所述解码器对所述自注意力结果以及所述编码数据进行空间位置注意力处理，得到解码数据的步骤，包括：根据所述自注意力结果，确定第三查询特征；根据所述编码数据，确定第三键特征和第三值特征；通过所述解码器对所述第三查询特征、所述第三键特征和所述第三值特征进行空间位置注意力处理，得到所述解码数据。11.一种目标检测装置，其特征在于，所述装置包括：编码数据确定模块，用于根据所述待检测图像的特征张量确定第一查询特征、第一键特征和第一值特征；通过所述编码器对所述第一查询特征、所述第一键特征和所述第一值特征进行空间位置自注意力处理，得到空间位置自注意力结果；根据所述空间位置自注意力结果，确定所述编码器输出的编码数据；其中，所述空间位置自注意力处理为将所述第一键特征进行行列解耦，并根据行列解耦得到的第一键行特征、第一键列特征以及所述第一查询特征和所述第一值特征，确定所述空间位置自注意力结果；解码数据确定模块，用于通过所述解码器对第二查询特征、第二键特征、第二值特征以及所述编码数据进行处理，得到解码数据，其中，所述第二查询特征、所述第二键特征和所述第二值特征通过预设参数确定；检测结果确定模块，用于根据所述解码数据，确定所述待检测图像的检测结果，所述检测结果包括所述待检测图像中的目标对象对应的目标特征集。12.一种电子设备，其特征在于，包括处理器和存储器，所述存储器存储有能够被所述处理器执行的计算机可执行指令，所述处理器执行所述计算机可执行指令以实现权利要求1至10任一项所述的方法。
13.一种计算机可读存储介质，其特征在于，所述计算机可读存储介质存储有计算机可执行指令，所述计算机可执行指令在被处理器调用和执行时，计算机可执行指令促使处理器实现权利要求1至10任一项所述的方法。

技术总结
本发明提供了一种目标检测方法、装置及电子设备，该方法包括：根据待检测图像的特征张量确定第一查询特征、第一键特征和第一值特征；通过编码器对第一查询特征、第一键特征和第一值特征进行空间位置自注意力处理，得到空间位置自注意力结果；根据空间位置自注意力结果，确定编码器输出的编码数据；通过解码器对第二查询特征、第二键特征、第二值特征以及编码数据进行处理，得到解码数据；根据解码数据，确定待检测图像对应的检测结果。本发明通过对第一键特征进行行列解耦，并基于行列解耦得到的结果，分别在行和列进行自注意力处理，能够在保证目标检测的准确性的基础上，有效降低显存消耗。存消耗。存消耗。


技术研发人员：王英明
受保护的技术使用者：北京迈格威科技有限公司
技术研发日：2021.06.11
技术公布日：2021/10/29