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

说明书

本申请实施例提供了一种目标检测方法、装置、电子设备及存储介质。其中，目标检测方法包括：读取待检测的第一个视频帧作为检测帧，对所述检测帧进行目标检测，得到所述检测帧中检测框的检测信息；从待检测的第二个视频帧开始，连续读取两个视频帧分别作为预测帧和检测帧，对所述检测帧进行目标检测，得到所述检测帧中检测框的检测信息；基于所述预测帧的前一检测帧中检测框的检测信息和所述预测帧的后一检测帧中检测框的检测信息进行匹配，确定所述前一检测帧中匹配成功的第一检测框；预测所述第一检测框在所述预测帧中对应的预测框的预测信息。本申请实施例中，能够降低性能消耗，提高检测效率，进而满足视频目标检测的实时性要求。

技术领域

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

背景技术

视频目标检测是图像目标检测在视频领域的自然延伸，因为视频的本质还是连续的图像，因此视频目标检测的基本原理与图像目标检测类似。视频目标检测是自主驾驶感知、监控、可穿戴设备、物联网等应用场景中的一项重要任务。

现有技术中，视频目标检测的方式是，对视频中的每个视频帧逐帧进行目标检测。但是由于视频中视频帧的数量很大，因此逐帧进行目标检测的方式导致性能消耗较大，检测效率较低。

发明内容

鉴于上述问题，本申请实施例提出了一种目标检测方法、装置、电子设备及存储介质，用以解决逐帧进行目标检测的方式导致性能消耗较大，检测效率较低的问题。

根据本申请的实施例的第一方面，提供了一种目标检测方法，包括：

读取待检测的第一个视频帧作为检测帧，对所述检测帧进行目标检测，得到所述检测帧中检测框的检测信息；

从待检测的第二个视频帧开始，连续读取两个视频帧分别作为预测帧和检测帧，对所述检测帧进行目标检测，得到所述检测帧中检测框的检测信息；

基于所述预测帧的前一检测帧中检测框的检测信息和所述预测帧的后一检测帧中检测框的检测信息进行匹配，确定所述前一检测帧中匹配成功的第一检测框；

预测所述第一检测框在所述预测帧中对应的预测框的预测信息。

可选地，所述基于所述预测帧的前一检测帧中检测框的检测信息和所述预测帧的后一检测帧中检测框的检测信息进行匹配，确定所述前一检测帧中匹配成功的第一检测框，包括：针对所述前一检测帧中的每个检测框，基于当前检测框的检测信息，确定所述当前检测框在所述后一检测帧中的预测信息；将所述当前检测框的预测信息和所述后一检测帧中检测框的检测信息进行匹配；在所述当前检测框匹配成功时，将所述当前检测框确定为所述前一检测帧中匹配成功的第一检测框。

可选地，所述基于当前检测框的检测信息，确定所述当前检测框的在所述后一检测帧中预测信息，包括：将所述当前检测框的检测信息中包含的检测中心点位置作为角点，基于所述前一检测帧和所述预测帧进行光流检测，得到所述检测中心点位置在所述预测帧中的第一预测中心点位置；计算所述第一预测中心点位置与所述检测中心点位置之间的偏差信息；基于所述偏差信息计算所述检测中心点位置在所述后一检测帧中的第二预测中心点位置；将所述当前检测框的检测信息中包含的检测尺寸信息和所述第二预测中心点位置，确定为所述当前检测框在所述后一检测帧中的预测信息。

可选地，所述将所述当前检测框的预测信息和所述后一检测帧中检测框的检测信息进行匹配，包括：基于所述当前检测框的预测信息和所述后一检测帧中检测框的检测信息，计算所述当前检测框与所述后一检测帧中检测框的交并比；在所述交并比大于第一预设阈值时，确定所述当前检测框匹配成功。

可选地，在所述计算所述当前检测框与所述后一检测帧中的检测框的交并比之后，还包括：在所述交并比小于等于第一预设阈值时，基于所述当前检测框的预测信息和所述后一检测帧中检测框的检测信息，计算所述当前检测框与所述后一检测帧中检测框的距离；在所述距离小于第二预设阈值时，确定所述当前检测框匹配成功。