[发明专利]小目标对象检测方法、装置、电子设备及存储介质

说明书

本申请提供一种小目标对象检测方法、装置、电子设备及存储介质。该方法包括：获取原始输入图像，利用分辨率检测模型对原始输入图像进行检测，以判断原始输入图像是否满足预设分辨率要求；当原始输入图像不满足预设分辨率要求时，利用训练后的超分辨率增强模型对原始输入图像执行图像增强操作，输出符合分辨率要求的目标图像；将目标图像输入到小目标对象检测模型中，利用小目标对象检测模型对目标图像中的小目标对象进行定位和识别，得到每个小目标对象的位置信息以及数量信息。本申请提高远距离拍摄场景下的小目标对象检测的准确性和鲁棒性。

技术领域

本申请涉及计算机技术领域，尤其涉及一种小目标对象检测方法、装置、电子设备及存储介质。

背景技术

小目标检测是目标检测的一个子领域，主要是对具有特定特征的目标进行检测，有体积小、易遮挡、数据难获取等较为明显的领域特征，在多种实际场景下具有很大的理论和实践意义，例如：监视、无人机场景分析、自动行人检测、交通标志检测等场景。

现有技术中，尽管人体检测算法已经在上述场景中取得了一些实质性的进展，但是在一些特殊场景下，仍然无法达到预期的识别效果，例如以室外塔吊拍摄场景为例，由于室外塔吊摄像头的分辨率普遍不高，且拍摄距离越远人体目标越小，导致目前人体检测技术无法识别室外塔吊摄像头所拍摄到的人体目标；另外，由于在作业面上会有各种钢筋以及作业器械对人体造成遮挡，因此给小目标人体检测添加了更多的干扰。

发明内容

有鉴于此，本申请实施例提供了一种小目标对象检测方法、装置、电子设备及存储介质，以解决现有技术存在的针对室外塔吊拍摄场景无法准确识别小目标对象的问题。

本申请实施例的第一方面，提供了一种小目标对象检测方法，包括：获取原始输入图像，利用分辨率检测模型对原始输入图像进行检测，以判断原始输入图像是否满足预设分辨率要求；当原始输入图像不满足预设分辨率要求时，利用训练后的超分辨率增强模型对原始输入图像执行图像增强操作，输出符合分辨率要求的目标图像；将目标图像输入到小目标对象检测模型中，利用小目标对象检测模型对目标图像中的小目标对象进行定位和识别，得到每个小目标对象的位置信息以及数量信息。

本申请实施例的第二方面，提供了一种小目标对象检测装置，包括：分辨率检测模块，被配置为获取原始输入图像，利用分辨率检测模型对原始输入图像进行检测，以判断原始输入图像是否满足预设分辨率要求；超分辨率增强模块，被配置为当原始输入图像不满足预设分辨率要求时，利用训练后的超分辨率增强模型对原始输入图像执行图像增强操作，输出符合分辨率要求的目标图像；小目标对象检测模块，被配置为将目标图像输入到小目标对象检测模型中，利用小目标对象检测模型对目标图像中的小目标对象进行定位和识别，得到每个小目标对象的位置信息以及数量信息。

本申请实施例的第三方面，提供了一种电子设备，包括存储器，处理器及存储在存储器上并可在处理器上运行的计算机程序，处理器执行程序时实现上述方法的步骤。

本申请实施例的第四方面，提供了一种计算机可读存储介质，该计算机可读存储介质存储有计算机程序，该计算机程序被处理器执行时实现上述方法的步骤。

本申请实施例采用的上述至少一个技术方案能够达到以下有益效果：

通过获取原始输入图像，利用分辨率检测模型对原始输入图像进行检测，以判断原始输入图像是否满足预设分辨率要求；当原始输入图像不满足预设分辨率要求时，利用训练后的超分辨率增强模型对原始输入图像执行图像增强操作，输出符合分辨率要求的目标图像；将目标图像输入到小目标对象检测模型中，利用小目标对象检测模型对目标图像中的小目标对象进行定位和识别，得到每个小目标对象的位置信息以及数量信息。本申请实现室外塔吊拍摄场景下的小目标对象检测任务，能够准确识别出室外塔吊摄像头图像中特定的小目标对象（例如人体对象），提高远距离拍摄场景下的小目标对象检测的准确性和鲁棒性。

附图说明