[发明专利]一种目标落地检测方法、装置、电子设备和存储介质

说明书

本发明实施例公开了一种目标落地检测方法、装置、电子设备和存储介质，所述方法包括：确定原始图像中检测目标对应的第一候选框；对第一候选框的大小进行扩展，获得第二候选框，以使第二候选框包含检测目标周围环境的环境图像；基于空间注意力机制根据第二候选框内的图像确定检测目标是否位于地面上。通过本发明实施例的技术方案，不仅实现了端到端的落地检测，还节省了人工标注工作量，降低了计算复杂度，提高了落地检测的实时性与准确性。

技术领域

本发明实施例涉及目标检测与识别技术领域，尤其涉及一种目标落地检测方法、装置、电子设备和存储介质。

背景技术

随着人工智能技术的迅猛发展，人们的生活正在朝着更加智能化的方向前进。视频监控作为安全生产系统的重要组成部分，借助人工智能、深度学习等技术，能够实时监控分析一些场景下的各种事件，有效降低各类不合规事件的发生。在物流站点，每天都会有数量巨大的包裹发送，由于缺乏有效的监控手段，大量的软包裹(包括信封件)被随意散落在地面，直接和地面接触，导致大量的软包裹损坏。不仅给客户带来较差的体验，还给物流公司造成巨大的经济损失。因此需要一种可以有效检测软包裹是否落地的方法来有效监控上述事件的发生，及时采取措施将其放到规范的地方以避免在地面上被损坏。

目前常用的检测方法是：首先通过图像分割技术将软包裹和地面分割出来，然后设计额外的判定逻辑判定软包裹是否被放置在地面。

在实现本发明过程中，发明人发现现有技术中至少存在如下问题：

图像分割需要大量的以像素为单位的标注样本，像素级的标注工作量庞大，人力成本高；图像分割算法中涉及到的编码网络和解码网络的复杂度较高，计算量大，无法满足实时性的要求；对于难以分辨的类别错误率较高，例如箱体上的面单和信封件非常相似，难以分辨；需要额外的判定逻辑判定软包裹是否被放置在地面，无法实现端到端的判定。

发明内容

本发明实施例提供了一种目标落地检测方法、装置、电子设备和存储介质，不仅实现了端到端的落地检测，还节省了人工标注工作量，降低了计算复杂度，提高了落地检测的实时性与准确性。

第一方面，本发明实施例提供了一种目标落地检测方法，该方法包括：

确定原始图像中检测目标对应的第一候选框；

对所述第一候选框的大小进行扩展，获得第二候选框，以使所述第二候选框包含所述检测目标周围环境的环境图像；

基于空间注意力机制根据所述第二候选框内的图像确定所述检测目标是否位于地面上。

第二方面，本发明实施例还提供了一种目标落地检测装置，该装置包括：

确定模块，用于确定原始图像中检测目标对应的第一候选框；

扩展模块，用于对所述第一候选框的大小进行扩展，获得第二候选框，以使所述第二候选框包含所述检测目标周围环境的环境图像；

检测模块，用于基于空间注意力机制根据所述第二候选框内的图像确定所述检测目标是否位于地面上。

第三方面，本发明实施例还提供了一种电子设备，所述电子设备包括：

一个或多个处理器；

存储器，用于存储一个或多个程序；

当所述一个或多个程序被所述一个或多个处理器执行，使得所述一个或多个处理器实现如本发明任意实施例所提供的目标落地检测方法步骤以及步骤。

第四方面，本发明实施例还提供了一种计算机可读存储介质，其上存储有计算机程序，该程序被处理器执行时实现如本发明任意实施例所提供的目标落地检测方法步骤以及步骤。