目标对象的预测方法、装置、电子设备和存储介质与流程

技术特征：
1.一种目标对象的预测方法，其特征在于，所述方法包括：获取待处理的原始图像，所述原始图像包含至少一个目标对象；将所述原始图像输入至训练完毕的第一检测模型，输出至少一个所述目标对象的第一位置信息和粗略类别信息；针对每个所述目标对象，根据所述第一位置信息对所述原始图像进行裁剪得到多个目标对象区域图像，将多个所述目标对象区域图像输入至训练完毕的第二检测模型，输出每个所述目标对象的多个第二位置信息和多个精细类别信息；针对每个所述目标对象，根据所述粗略类别信息、所述第一位置信息、多个所述第二位置信息和预设交并比阈值，判断每个所述目标对象是否属于预设类别的目标类别对象；若至少一个所述目标对象中存在所述目标类别对象，则计算所述目标类别对象的置信度，并根据所述置信度从所述目标类别对象中选择出目标预测对象；根据所述目标预测对象的所述第一位置信息和所述第二位置信息生成所述目标预测对象的位置预测结果，并将所述目标预测对象对应的所述精细类别信息作为类别预测结果。2.根据权利要求1所述的方法，其特征在于，所述根据所述粗略类别信息、所述第一位置信息、多个所述第二位置信息和预设交并比阈值，判断每个所述目标对象是否属于预设类别的目标类别对象，包括：当所述粗略类别信息属于所述预设类别时，根据所述第一位置信息、多个所述第二位置信息和预设交并比阈值，判断所述目标对象是否位于多个所述目标对象区域图像的中心区域；将位于至少一个所述目标对象区域图像的中心区域的所述目标对象确定为所述目标类别对象。3.根据权利要求2所述的方法，其特征在于，所述根据所述第一位置信息、多个所述第二位置信息和预设交并比阈值，判断所述目标对象是否位于多个所述目标对象区域图像的中心区域，包括：根据所述第一位置信息和多个所述第二位置信息计算所述目标对象的多个交并比参数；若存在至少一个大于或等于所述预设交并比阈值的交并比参数，则确认所述目标对象位于至少一个所述目标对象区域图像的中心区域。4.根据权利要求2或3所述的方法，其特征在于，所述计算所述目标类别对象的置信度，包括：统计所述目标类别对象位于所述中心区域的数量，根据所述数量和包含所述目标类别对象的所述目标对象区域图像的数量计算所述置信度。5.根据权利要求1所述的方法，其特征在于，所述根据所述置信度从所述目标类别对象中选择出目标预测对象，包括：从所述目标类别对象中选择所述置信度满足预设条件的所述目标预测对象。6.根据权利要求1所述的方法，其特征在于，所述根据所述目标预测对象的所述第一位置信息和所述第二位置信息生成所述目标预测对象的位置预测结果，包括：计算所述第一位置信息和所述第二位置信息的平均值，将所述平均值作为所述位置预
测结果。7.根据权利要求1所述的方法，其特征在于，所述第一检测模型包含two-stage网络模型，所述第二检测模型包含one-stage网络模型。8.一种目标对象的预测装置，其特征在于，所述装置包括：图像获取模块，用于获取待处理的原始图像，所述原始图像包含至少一个目标对象；第一检测模块，用于将所述原始图像输入至训练完毕的第一检测模型，输出至少一个所述目标对象的第一位置信息和粗略类别信息；第二检测模块，用于针对每个所述目标对象，根据所述第一位置信息对所述原始图像进行裁剪得到多个目标对象区域图像，将多个所述目标对象区域图像输入至训练完毕的第二检测模型，输出每个所述目标对象的多个第二位置信息和多个精细类别信息；对象判断模块，用于针对每个所述目标对象，根据所述粗略类别信息、所述第一位置信息、多个所述第二位置信息和预设交并比阈值，判断每个所述目标对象是否属于预设类别的目标类别对象；对象选择模块，用于若至少一个所述目标对象中存在所述目标类别对象，则计算所述目标类别对象的置信度，并根据所述置信度从所述目标类别对象中选择出目标预测对象；结果确定模块，用于根据所述目标预测对象的所述第一位置信息和所述第二位置信息生成所述目标预测对象的位置预测结果，并将所述目标预测对象对应的所述精细类别信息作为类别预测结果。9.一种电子设备，包括存储器、处理器及存储在所述存储器上并可在处理器上运行的计算机程序，其特征在于，所述处理器执行所述计算机程序时实现权利要求1至7中任意一项所述的目标对象的预测方法。10.一种计算机可读存储介质，其上存储有计算机程序，其特征在于，该程序被处理器执行时实现权利要求1至7中任意一项所述的目标对象的预测方法。

技术总结
本发明实施例提供了一种目标对象的预测方法和装置，其中，所述方法包括：将原始图像输入至第一检测模型，输出目标对象的第一位置信息和粗略类别信息；对原始图像进行裁剪得到多个目标对象区域图像，将多个目标对象区域图像输入至第二检测模型，输出每个目标对象的多个第二位置信息和多个精细类别信息；若存在目标类别对象则选择出目标预测对象；根据第一位置信息和第二位置信息生成位置预测结果，将精细类别信息作为类别预测结果。本发明实施例增加了目标对象的上下文信息，从而提高从第二检测模型输出的精细类别信息的准确率。针对目标类别对象选择出目标预测对象，避免了非目标预测对象的干扰，进一步提升目标对象预测的准确率。率。率。


技术研发人员：张珂 罗钧峰 苏金明 范铭源 魏晓明 魏晓林
受保护的技术使用者：北京三快在线科技有限公司
技术研发日：2021.12.24
技术公布日：2022/4/29