[发明专利]目标检测模型训练方法、目标检测方法、装置和电子设备

说明书

本公开提供了一种目标检测模型训练方法、目标检测方法、装置和电子设备，涉及人工智能技术领域，具体为深度学习、图像处理、计算机视觉技术领域。该方法包括：在教师模型中，根据样本图像得到教师序列特征；将教师查询向量和教师序列特征输入解码器得到增强教师查询向量，并根据增强教师查询向量进行预测得到教师预测信息；在学生模型中，根据样本图像得到学生序列特征；将学生查询向量和学生序列特征输入解码器得到增强学生查询向量，并根据增强学生查询向量进行预测得到学生预测信息；根据教师预测信息、学生预测信息和样本图像的真值信息，对教师查询向量进行训练，并对学生模型进行训练。通过上述技术方案能够提高目标检测性能。

技术领域

本公开涉及人工智能技术领域，具体为深度学习、图像处理、计算机视觉技术领域。具体涉及一种目标检测模型训练方法、目标检测方法、装置和电子设备。

背景技术

随着人工智能(Artificial Intelligence，AI)的快速发展，人工智能已经广泛应用于计算机视觉技术、语音识别技术、自然语言处理技术、深度学习、大数据处理技术等领域。

基于人工智能的图像检测技术也已经广泛应用于不同场景。其中，基于人工智能的目标检测技术，用于确定图像中的物体边界框和物体类别。如何提高目标检测性能十分重要。

发明内容

本公开提供了一种目标检测模型训练方法、目标检测方法、装置和电子设备。

根据本公开的一方面，提供了一种目标检测模型训练方法，包括：

在教师模型中，将样本图像输入图像特征提取网络得到教师图像特征，并将所述教师图像特征输入编码器得到教师序列特征；将可学习的教师查询向量和所述教师序列特征输入解码器得到增强教师查询向量，并根据所述增强教师查询向量进行预测得到教师预测信息；

在学生模型中，将所述样本图像输入图像特征提取网络得到学生图像特征，并将所述学生图像特征输入编码器得到学生序列特征；将可学习的学生查询向量和所述学生序列特征输入解码器得到增强学生查询向量，并根据所述增强学生查询向量进行预测得到学生预测信息；

根据所述教师预测信息、所述学生预测信息和样本图像的真值信息，对所述教师查询向量进行训练，并对所述学生模型进行训练。

根据本公开的另一方面，提供了一种目标检测方法，包括：

获取待检测的目标图像；

将所述目标图像输入目标检测模型得到检测结果；所述目标检测模型为采用本公开任意实施例公开的目标检测模型训练方法所训练得到的学生模型。

根据本公开的又一方面，提供了一种目标检测模型训练装置，包括：

教师预测模块，用于在教师模型中，将样本图像输入图像特征提取网络得到教师图像特征，并将所述教师图像特征输入编码器得到教师序列特征；将可学习的教师查询向量和所述教师序列特征输入解码器得到增强教师查询向量，并根据所述增强教师查询向量进行预测得到教师预测信息；

学生预测模块，用于在学生模型中，将所述样本图像输入图像特征提取网络得到学生图像特征，并将所述学生图像特征输入编码器得到学生序列特征；将可学习的学生查询向量和所述学生序列特征输入解码器得到增强学生查询向量，并根据所述增强学生查询向量进行预测得到学生预测信息；

模型训练模块，用于根据所述教师预测信息、所述学生预测信息和样本图像的真值信息，对所述教师查询向量进行训练，并对所述学生模型进行训练。

根据本公开的又一方面，提供了一种目标检测装置，包括：

目标图像模块，用于获取待检测的目标图像；