[发明专利]SAR图像目标检测方法、装置、电子设备和存储介质

权利要求书

1.一种SAR图像目标检测方法，其特征在于，包括：

将原始SAR图像输入至特征提取模型，得到所述特征提取模型输出的多个预测特征图；各预测特征图的分辨率不同，且所述多个预测特征图中至少包括一个分辨率大于预设值的特征图；

基于预设锚点，以及各预测特征图的分辨率，确定各预测特征图对应的目标对象候选边界框以及各目标对象候选边界框对应的类别置信度；所述类别置信度为各目标对象候选边界框存在目标对象的概率；

基于非极大抑制算法，以及各目标对象候选边界框对应的类别置信度，从所有目标对象候选边界框中确定目标边界框，并将所述目标边界框作为所述SAR图像的目标检测结果；

其中，所述特征提取模型是基于样本SAR图像训练得到的；所述特征提取模型用于提取所述原始SAR图像的多个具有不同分辨率的初始特征，并按照分辨率递增的顺序依次对各初始特征进行融合，得到对应的初始融合特征后，按照分辨率递减的顺序依次对各初始融合特征进行融合，得到多个预测特征图。

2.根据权利要求1所述的SAR图像目标检测方法，其特征在于，各初始融合特征是基于如下步骤获取的：

第一卷积步骤：按照分辨率递增的顺序，对各初始特征进行排序，并对第一个初始特征进行卷积操作，得到第一初始融合特征；

第一融合步骤：对所述第一初始融合特征进行上采样后，与所述第一初始融合特征对应的下一初始特征进行融合，得到第二初始融合特征；其中，上采样后的第一初始融合特征与所述第一初始融合特征对应的下一初始特征的维度相同；

第一循环步骤：将所述第二初始融合特征作为所述第一初始融合特征，循环执行所述第一融合步骤，直至所述第一初始融合特征不存在对应的下一初始特征。

3.根据权利要求1所述的SAR图像目标检测方法，其特征在于，各预测特征图是基于如下步骤获取的：

第二卷积步骤：按照分辨率递减的顺序，对各初始融合特征进行排序，并对第一个初始融合特征进行卷积操作，得到第一预测特征图；

第二融合步骤：对所述第一预测特征图进行下采样后，与所述第一预测特征图对应的下一初始融合特征进行融合，得到第二预测特征图；其中，下采样后的第一预测特征图与所述第一预测特征图对应的下一初始融合特征的维度相同；

第二循环步骤：将所述第二预测特征图作为所述第一预测特征图，循环执行所述第二融合步骤，直至所述第一预测特征图不存在对应的下一初始融合特征。

4.根据权利要求1所述的SAR图像目标检测方法，其特征在于，所述预设锚点是基于如下步骤获取的：

获取多个SAR训练集图像，并确定各SAR训练集图像中目标对象的边界框；

按照预设簇数，对各SAR训练集图像中目标对象的边界框进行尺寸聚类，并将聚类后得到的各簇中心作为所述预设锚点中的各锚点。

5.根据权利要求1所述的SAR图像目标检测方法，其特征在于，所述基于预设锚点，以及各预测特征图的分辨率，确定各预测特征图对应的目标对象候选边界框以及各目标对象候选边界框对应的类别置信度，包括：

基于所述预设锚点中各锚点的面积，以及各预测特征图的分辨率，从所述预设锚点中获取各预测特征图的目标锚点；

将各目标锚点在对应的预测特征图上进行平移和缩放，确定各预测特征图对应的目标对象候选边界框以及各目标对象候选边界框对应的类别置信度。

6.根据权利要求5所述的SAR图像目标检测方法，其特征在于，所述基于所述预设锚点中各锚点的面积，以及各预测特征图的分辨率，从所述预设锚点中获取各预测特征图的目标锚点，包括：

将所述预设锚点中各锚点按照面积从小到大排序，基于预测特征图的数量对排序后的各锚点进行平均分组，并将各分组的锚点作为所述目标锚点分配至对应的预测特征图；

其中，预测特征图的分辨率越高，预测特征图对应的目标锚点面积越小。

7.根据权利要求1至6任一项所述的SAR图像目标检测方法，其特征在于，各初始特征的尺度分别为所述原始SAR图像尺度的1/4、1/8、1/16以及1/32。