[发明专利]基于场景特性学习的半监督SAR舰船检测方法

权利要求书

1.一种基于场景特性学习的半监督SAR舰船检测方法，其特征在于，包括如下步骤：

(1)生成训练集：

采集至少21张大幅SAR图像，将每张大幅SAR图像裁剪为512×512大小的多张子图像；从包含舰船目标的子图像中随机挑选30％进行目标级标记和场景级标记，其余的子图像只进行场景级标记，将所有标记后的子图像组成训练集；

(2)构建半监督SAR舰船检测网络：

(2a)搭建一个由八个卷积块串联组成的特征提取子网络；

(2b)搭建一个由四个卷积块和四个检测头组成的检测子网络，其中四个卷积块首先依次串联连接，然后每个卷积块再分别与其对应的检测头连接；

(2c)搭建由场景识别模块和场景聚合模块并联组成的场景特性学习子网络；

(2d)将场景特性学习子网络与检测子网络并联，然后将它们与特征提取子网络串联构成半监督SAR舰船检测网络；

(3)将训练集输入到半监督SAR舰船检测网络中，利用随机梯度下降算法，迭代更新网络的权重值，优化网络的总损失函数直至其收敛，得到训练好的半监督SAR舰船检测网络；

(4)检测待测试SAR图像中的目标框位置：

(4a)将待测试的大幅SAR图像滑窗裁剪为多张512×512大小的子图像；

(4b)将每张测试子图像依次输入到训练好的特征提取子网络和场景识别模块中，得到测试子图像的场景识别结果；

(4c)根据场景识别结果，得出目标框位置和目标框类别：

对于场景识别结果为内陆场景的测试子图像，输出检测结果为无目标；

对于场景识别结果为近岸场景的测试子图像，将该测试子图像输入到检测子网络中，并设置近岸检测阈值th_in，得到该测试子图像的目标框位置和目标框类别；

对于场景识别结果为远海场景的测试子图像，将该测试子图像输入到检测子网络中，并设置远海检测阈值th_off，得到该测试子图像的目标框位置和目标框类别；

(5)依照滑窗的顺序，将每张测试子图像的目标框位置映射到待测试的每张大幅SAR图像的对应位置上，得到大幅SAR图像的舰船检测结果。

2.根据权利要求1所述的方法，其特征在于：步骤(2a)搭建的特征提取子网络中，各卷积块的结构参数如下：

第一卷积块中包含卷积层和池化层，其中卷积层个数设置为1，该卷积层的卷积核大小设置为7×7，池化层个数设置为1，池化区域大小设置为2×2；

第二卷积块的卷积层个数设置为4，该4个卷积层的卷积核大小分别设置为1×1，3×3，1×1，1×1；

第三卷积块的卷积层个数设置为3，该3个卷积层的卷积核大小分别设置为1×1，3×3，1×1；

第四卷积块、第六卷积块、第七卷积块、第八卷积块的卷积层个数和卷积核大小均与第三卷积块相同；

第五卷积块的卷积层个数和卷积核大小与第二卷积块相同。

3.根据权利要求1所述的方法，其特征在于：步骤(2b)搭建的检测子网络中，各卷积块的结构参数如下：

第一卷积块的卷积层个数设置为10，该10个卷积层的卷积核大小分别设置为1×1，3×3，1×1，1×1，1×1，3×3，1×1，1×1，3×3，1×1；

第二卷积块的卷积层个数设置为9，该9个卷积层的卷积核大小分别设置为1×1，3×3，1×1，1×1，3×3，1×1，1×1，3×3，1×1；

第三卷积块的卷积层个数设置为10，该10个卷积层的卷积核大小分别设置为1×1，3×3，1×1，1×1，1×1，3×3，1×1，1×1，3×3，1×1；

第四卷积块的卷积层个数设置为4，该4个卷积层的卷积核大小分别设置为1×1，3×3，1×1，1×1；

在四个检测头中，每个检测头都包含并列的两个卷积层，每个卷积核的大小均为3×3。