[发明专利]目标检测方法、目标检测模型的训练方法及装置

说明书

本公开提供了一种目标检测方法、目标检测模型的训练方法及装置，涉及人工智能技术领域。方法包括：获取待检测图像；将待检测图像输入预先训练的目标检测模型，得到待检测图像对应的目标检测结果；在目标检测结果满足预设条件的情况下，获取目标检测结果中的位置框信息；其中，预设条件为目标检测结果中包含位置框信息，且目标检测结果对应的置信度小于预设的置信度阈值；基于位置框信息对目标检测结果进行校验，得到待检测图像对应的目标校验结果。本公开技术方案中，在目标检测模型输出的目标检测结果中，对于包含位置框信息，且置信度小于预设的置信度阈值的目标检测结果进行校验，以提升图像审核的准确率和包含目标对象的图像的召回率。

技术领域

本公开涉及人工智能技术领域，尤其涉及机器学习领域。

背景技术

随着电子商务的兴起，商家广告宣传手段日益多样，丰富的图片宣传为广告宣传的重要手段。现实广告物料推广过程中，存在商家在图文宣传过程中，违规发布包含特定目标的图像的情况，需要对商家广告物料中提供的图片进行审核，现有技术中的内容审核方案准确率较低，无法满足实际需要。

发明内容

本公开提供了一种目标检测方法、目标检测模型的训练方法、装置、设备以及存储介质。

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

获取待检测图像；

将待检测图像输入预先训练的目标检测模型，得到待检测图像对应的目标检测结果；

在目标检测结果满足预设条件的情况下，获取目标检测结果中的位置框信息；其中，预设条件为目标检测结果中包含位置框信息，且目标检测结果对应的置信度小于预设的置信度阈值；

基于位置框信息对目标检测结果进行校验，得到待检测图像对应的目标校验结果。

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

获取多组第一训练样本数据，基于多组第一训练样本数据对初始目标检测模型进行训练，直到满足预设的训练结束条件，得到本公开任一实施例的目标检测模型；

其中，第一训练样本数据包括第一样本图像以及第一样本图像对应的第一样本标签，第一样本标签用于表征第一样本图像中是否包含目标对象以及目标对象的位置框信息，第一样本图像包括拼接图像；拼接图像是基于目标对象图像和相似目标图像进行拼接得到的，目标对象图像为包含目标对象的图像，相似目标图像是包含相似目标对象的图像，相似目标对象与目标对象的相似度在预设范围内。

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

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

目标检测模块，用于将待检测图像输入预先训练的目标检测模型，得到待检测图像对应的目标检测结果；

位置框获取模块，用于在目标检测结果满足预设条件的情况下，获取目标检测结果中的位置框信息；其中，预设条件为目标检测结果中包含位置框信息，且目标检测结果对应的置信度小于预设的置信度阈值；

结果校验模块，用于基于位置框信息对目标检测结果进行校验，得到待检测图像对应的目标校验结果。

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

样本获取模块，用于获取多组第一训练样本数据；

模型训练模块，用于基于多组第一训练样本数据对初始目标检测模型进行训练，直到满足预设的训练结束条件，得到本公开任一实施例的目标检测模型；