[发明专利]船舶检测方法、装置、电子设备和计算机可读介质

说明书

本公开的实施例公开了船舶检测方法、装置、电子设备和计算机可读介质。该方法的一具体实施方式包括：获取目标图像；将目标图像输入至预先训练的船舶检测模型，以生成第一候选图像和目标类别信息；将第一候选图像中感兴趣框框定的子图像，确定为第二候选图像；基于第二候选图像和预先训练的目标语义分割网络，生成船舶区域图像和船舶区域图像对应的语义信息；根据语义信息，确定船舶区域图像包括的船舶的行驶方向，以生成行驶方向信息；将目标类别信息和行驶方向信息发送至目标终端以供显示。该实施方式提高了船舶检测的成功率。

技术领域

本公开的实施例涉及计算机技术领域，具体涉及船舶检测方法、装置、电子设备和计算机可读介质。

背景技术

船舶检测，是用于识别出图像中包含的船舶的一项技术。目前，在进行船舶检测时，通常采用的方式为：通过卷积神经网络，直接对图像上的船舶进行检测。

然而，当采用上述方式时，经常会存在如下技术问题：

第一，当被检测图像分辨率较低或被检测船舶的尺度较小、特征较少时，往往会导致船舶检测的成功率低。

第二，目前的卷积神经网络在对图像进行检测时往往会检测出船舶的位置和类别，往往不能检测出被检测船舶的方向信息。

发明内容

本公开的内容部分用于以简要的形式介绍构思，这些构思将在后面的具体实施方式部分被详细描述。本公开的内容部分并不旨在标识要求保护的技术方案的关键特征或必要特征，也不旨在用于限制所要求的保护的技术方案的范围。

本公开的一些实施例提出了船舶检测方法、装置、电子设备和计算机可读介质，来解决以上背景技术部分提到的技术问题中的一项或多项。

第一方面，本公开的一些实施例提供了一种船舶检测方法，该方法包括：获取目标图像，其中，上述目标图像为包含船舶的图像；将上述目标图像输入至预先训练的船舶检测模型，以生成第一候选图像和目标类别信息，其中，上述第一候选图像为包含感兴趣框的图像，上述目标类别信息为上述第一候选图像中感兴趣框框定的子图像对应的类别信息；将上述第一候选图像中感兴趣框框定的子图像，确定为第二候选图像；基于上述第二候选图像和预先训练的目标语义分割网络，生成船舶区域图像和上述船舶区域图像对应的语义信息，其中，上述船舶区域图像是从上述第二候选图像中包括的目标区域图像中分割出的图像，上述目标区域图像为包含船舶区域和船舶尾浪区域的图像；根据上述语义信息，确定船舶区域图像包括的船舶的行驶方向，以生成行驶方向信息；将上述目标类别信息和上述行驶方向信息发送至目标终端以供显示。

第二方面，本公开的一些实施例提供了一种船舶检测装置，装置包括：获取单元，被配置成获取目标图像，其中，上述目标图像为包含船舶的图像；输入单元，被配置成将上述目标图像输入至预先训练的船舶检测模型，以生成第一候选图像和目标类别信息，其中，上述第一候选图像为包含感兴趣框的图像，上述目标类别信息为上述第一候选图像中感兴趣框框定的子图像对应的类别信息；第一确定单元，被配置成将上述第一候选图像中感兴趣框框定的子图像，确定为第二候选图像；生成单元，被配置成基于上述第二候选图像和预先训练的目标语义分割网络，生成船舶区域图像和上述船舶区域图像对应的语义信息，其中，上述船舶区域图像是从上述第二候选图像中包括的目标区域图像中分割出的图像，上述目标区域图像为包含船舶区域和船舶尾浪区域的图像；第二确定单元，被配置成根据上述语义信息，确定船舶区域图像包括的船舶的行驶方向，以生成行驶方向信息；发送单元，被配置成将上述目标类别信息和上述行驶方向信息发送至目标终端以供显示。

第三方面，本公开的一些实施例提供了一种电子设备，包括：一个或多个处理器；存储装置，其上存储有一个或多个程序，当一个或多个程序被一个或多个处理器执行，使得一个或多个处理器实现上述第一方面任一实现方式所描述的方法。

第四方面，本公开的一些实施例提供了一种计算机可读介质，其上存储有计算机程序，其中，程序被处理器执行时实现上述第一方面任一实现方式所描述的方法。