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

权利要求书

1.一种船舶检测方法，包括：

获取目标图像，其中，所述目标图像为包含船舶的图像；

将所述目标图像输入至预先训练的船舶检测模型，以生成第一候选图像和目标类别信息，其中，所述第一候选图像为包含感兴趣框的图像，所述目标类别信息为所述第一候选图像中感兴趣框框定的子图像对应的类别信息；

将所述第一候选图像中感兴趣框框定的子图像，确定为第二候选图像；

基于所述第二候选图像和预先训练的目标语义分割网络，生成船舶区域图像和所述船舶区域图像对应的语义信息，其中，所述船舶区域图像是从所述第二候选图像中包括的目标区域图像中分割出的图像，所述目标区域图像为包含船舶区域和船舶尾浪区域的图像；

根据所述语义信息，确定船舶区域图像包括的船舶的行驶方向，以生成行驶方向信息；

将所述目标类别信息和所述行驶方向信息发送至目标终端以供显示。

2.根据权利要求1所述的方法，其中，所述方法还包括：

根据所述行驶方向信息，确定船舶是否偏离目标航道；

响应于船舶偏离所述目标航道，将预设提示信息发送至所述目标终端以供显示。

3.根据权利要求2所述的方法，其中，所述基于所述第二候选图像和预先训练的目标语义分割网络，生成船舶区域图像和所述船舶区域图像对应的语义信息，包括：

通过所述预先训练的目标语义分割网络，对所述第二候选图像进行像素级分类，以生成所述船舶区域图像和所述船舶区域图像对应的语义信息。

4.根据权利要求3所述的方法，其中，所述船舶区域图像对应的语义信息包括目标物位置，其中，所述目标物位置为船舶区域图像的中心位置。

5.根据权利要求4所述的方法，其中，所述根据所述语义信息，确定船舶区域图像包括的船舶的行驶方向，以生成行驶方向信息，包括：

对所述语义信息进行信号分析处理，得到所述船舶区域图像中的船舶区域的外包围框和船舶区域坐标系，其中，所述外包围框的中心位置为所述目标物位置；

将所述外包围框的坐标点中的特征坐标点发出的射线确定为目标射线，其中，所述目标射线的发出方向与所述船舶区域坐标系的横轴的正方向相同，所述目标射线与所述船舶区域坐标系的横轴平行；

基于所述目标射线，生成旋转角度；

根据所述旋转角度，生成行驶方向信息。

6.根据权利要求5所述的方法，其中，所述基于所述目标射线，生成旋转角度，包括：

将所述目标射线按照顺时针方向或者逆时针方向进行旋转；

响应于所述目标射线在旋转过程中与所述外包围框的至少一条边框中目标边框重合，将所述目标射线旋转的角度确定为所述旋转角度。

7.根据权利要求6所述的方法，其中，所述根据所述旋转角度，生成行驶方向信息，包括：

响应于所述旋转角度大于所述第一预设角度，将第二预设角度与所述旋转角度的差值确定为行驶方向信息。

8.一种船舶检测装置，包括：

获取单元，被配置成获取目标图像，其中，所述目标图像为包含船舶的图像；

输入单元，被配置成将所述目标图像输入至预先训练的船舶检测模型，以生成第一候选图像和目标类别信息，其中，所述第一候选图像为包含感兴趣框的图像，所述目标类别信息为所述第一候选图像中感兴趣框框定的子图像对应的类别信息；

第一确定单元，被配置成将所述第一候选图像中感兴趣框框定的子图像，确定为第二候选图像；

生成单元，被配置成基于所述第二候选图像和预先训练的目标语义分割网络，生成船舶区域图像和所述船舶区域图像对应的语义信息，其中，所述船舶区域图像是从所述第二候选图像中包括的目标区域图像中分割出的图像，所述目标区域图像为包含船舶区域和船舶尾浪区域的图像；

第二确定单元，被配置成根据所述语义信息，确定船舶区域图像包括的船舶的行驶方向，以生成行驶方向信息；

发送单元，被配置成将所述目标类别信息和所述行驶方向信息发送至目标终端以供显示。