[发明专利]一种小目标船只识别方法、装置及电子设备

权利要求书

1.一种小目标船只识别方法，其特征在于，包括：

获取船只数据；

根据所述船只数据进行图像放大预处理，获得预处理船只图像；其中，所述预处理船只图像包括：训练船只图像和验证船只图像；

基于所述训练船只图像输入目标检测识别网络，构建小目标船只识别模型；

通过向所述构建小目标船只识别模型输入所述验证船只图像进行检测，输出小目标船只识别结果。

2.根据权利要求1所述的方法，其特征在于，所述图像放大预处理包括：

基于所述获取船只数据进行小尺寸图像筛选；

利用拉普拉斯金字塔对所述筛选出的小尺寸图像进行上采样，以获得大尺寸图像。

3.根据权利要求2所述的方法，其特征在于，所述拉普拉斯金字塔图像为：

其中，G_i表示为原始尺寸是小尺寸图像的第i层，UP为上采样，g_5×5为5X5的高斯卷积核。

4.根据权利要求3所述的方法，其特征在于，所述目标检测识别网络，包括：

将所述训练船只图像输入卷积层中进行特征提取，输出船只图像的特征图；

将所述船只图像的特征图输入到RPN网络中提取船只图像候选区域；

利用ROI池化层提取所述船只图像候选区域的特征图；

将所述船只图像候选区域的特征图输入到Softmax层中进行分类及用于边框回归全连接层进行修正，输出小目标船只识别结果。

5.根据权利要求4所述的方法，其特征在于，所述构建小目标船只识别模型，包括：循环迭代目标识别检测网络直至边框回归，以获得小目标船只识别结果。

6.一种小目标船只识别装置，其他特征在于，包括：

获取模块，用于获取船只数据；

预处理模块，用于根据所述船只数据进行图像放大预处理，获得预处理船只图像；其中，所述预处理船只图像包括：训练船只图像和验证船只图像；

训练模块，用于基于所述训练船只图像输入目标检测识别网络，构建小目标船只识别模型；

识别模块，通过向所述构建小目标船只识别模型输入所述验证船只图像进行检测，输出小目标船只识别结果。

7.一种电子设备，其特征在于，包括：

存储器和处理器，所述存储器和所述处理器之间互相通信连接，所述存储器中存储有计算机指令，所述处理器通过执行所述计算机指令，从而执行权利要求1-5中任一项所述的小目标船只识别方法。

8.一种计算机可读存储介质，其特征在于，所述计算机可读存储介质存储有计算机指令，所述计算机指令用于使所述计算机执行权利要求1-5中任一项所述的小目标船只识别方法。