[发明专利]水上目标识别方法、装置、终端设备及存储介质

说明书

本申请适用于视觉检测技术领域，提供了一种水上目标识别方法、装置、终端设备及存储介质，所述水上目标识别方法包括：获取针对水上目标的视频流；将所述视频流中的参考视频帧输入至第一网络，得到所述参考视频帧中目标物体所在区域的图像，所述第一网络为剪枝处理后的YOLOv4网络，且所述YOLOv4网络的损失函数为Focal Loss；将所述目标物体所在区域的图像和所述视频流中所述参考视频帧的下一帧输入至第二网络，通过所述第二网络识别出所述参考视频帧的下一帧中的所述目标物体。通过本申请可解决现有目标识别方法容易造成漏检的问题。

技术领域

本申请属于视觉检测技术领域，尤其涉及一种水上目标识别方法、装置、终端设备及存储介质。

背景技术

传统的船只水面感知技术主要依靠船只上搭载的毫米波雷达、激光雷达和惯性测量单元、全球定位系统(Global Positioning System，GPS)等传感器。近年来计算机视觉技术发展迅速，视频流包含更丰富的目标物体细节信息，故基于视觉的感知技术更易于对水面上的目标物体进行有效地识别。

现有技术在对视频流进行目标识别时，通常是将每帧的目标识别视为独立事件，容易造成漏检。

发明内容

本申请实施例提供了一种水上目标识别方法、装置、终端设备及存储介质，以解决现有目标识别方法容易造成漏检的问题。

第一方面，本申请实施例提供了一种水上目标识别方法，所述水上目标识别方法包括：

获取针对水上目标物体的视频流；

将所述视频流中的参考视频帧输入至第一网络，得到所述参考视频帧中目标物体所在区域的图像，所述第一网络为剪枝处理后的YOLOv4网络，且所述YOLOv4网络的损失函数为Focal Loss；

将所述目标物体所在区域的图像和所述视频流中所述参考视频帧的下一帧输入至第二网络，通过所述第二网络识别出所述参考视频帧的下一帧中的所述目标物体。

第二方面，本申请实施例提供了一种水上目标识别装置，所述水上目标识别装置包括：

视频流获取模块，用于获取针对水上目标物体的视频流；

视频帧输入模块，用于将所述视频流中的参考视频帧输入至第一网络，得到所述参考视频帧中目标物体所在区域的图像，所述第一网络为剪枝处理后的YOLOv4网络，且所述YOLOv4网络的损失函数为Focal Loss；

目标识别模块，用于将所述目标物体所在区域的图像和所述视频流中所述参考视频帧的下一帧输入至第二网络，通过所述第二网络识别出所述参考视频帧的下一帧中的所述目标物体。

第三方面，本申请实施例提供了一种终端设备，包括存储器、处理器以及存储在所述存储器中并可在所述处理器上运行的计算机程序，所述处理器执行所述计算机程序时实现如上述第一方面所述水上目标识别方法的步骤。

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

第五方面，本申请实施例提供了一种计算机程序产品，当所述计算机程序产品在终端设备上运行时，使得所述终端设备执行如上述第一方面所述水上目标识别方法的步骤。