[发明专利]基于改进U-net网络的无人机麦田遥感图像分割方法

权利要求书

1.一种基于改进U-net网络的无人机麦田遥感图像分割方法，其特征在于它包括：

步骤1：以U-net为基础，融合多信息注意力模块，改善卷积在局部感受野上对特征图的无差别信息处理，加强对麦田和田埂相似特征差异的学习，达到分割精确化的目的；

步骤2：定义一个关注区域结构的混合损失函数，处理麦田图像中田埂与麦田的占比不均衡问题；

步骤3：利用形状先验信息，并结合阈值和矩形结构参数信息，以区域连通算法为指导对分割结果进一步修正。

2.根据权利要求1所述的方法，其特征在于步骤1中，多信息注意力模块包括空间注意力模块，和通道注意力模块；输入特征首先经过通道注意力模块，得到在通道维度上具有更强指向性和注意力信息的新特征；再将新特征输入空间注意力计算模块，最终得到融合空间和通道注意力信息的复合新特征。

3.根据权利要求2所述的方法，其特征在于通道注意力模块通过计算各通道的重要性并以此进行权重分配，得到融合通道注意力信息的新特征；其计算公式如下：

M_c＝σ(MLP(AvgPool(F))+MLP(MaxPool(F)))

式中，M_c表示获得通道注意力的新特征，σ表示sigmoid函数计算，MLP为特征的共享网络，AvgPool(F)表示平均池化特征，MaxPool(F)表示最大池化特征。

4.根据权利要求2所述的方法，其特征在于通过空间注意力模块捕获潜在信息，实现局部特征的丰富的上下文关系建立；计算公式如下：

式中，i和j为特征图中某像素点位置，N表示输出总数，A_j为融合通道注意力模块输出的新特征，特征A生成的三个新特征映射记做{B，C，D}∈R^C×N；α表示尺度系数。

5.根据权利要求1所述的方法，其特征在于步骤2中，提出基于区域结构的混合损失函数L_Rs为：

式中，N表示输出总数，η_i是每个损耗的权重，及分别表示第i个的焦点损失Focal Loss、结构相似损失SsimLoss和Tversky系数损失TverskyLoss。

6.根据权利要求1所述的方法，其特征在于步骤3中，修正步骤具体包括：

S3-1、利用尺寸为20×512的先验矩形模板遍历网络分割结果图像，对待补全田埂ROI进行初步选取，得到遴选的待补全田埂集合；

S3-2、对S3-1中遴选的待补全田埂集合进行筛选，得到待补全田埂ROI集合；

S3-3、取S3-2中待补全田埂ROI集合中某一ROI的中线中点p_center为种子节点；

S3-4、均分当前ROI为上下两部分，取二者中sum_p,xel较大者并以p_center为初始种子，利用区域连通算法得其宽度w′，并对当前ROI进行动态拟合得到新的待补全田埂ROI；

S3-5、对S3-4中新ROI进行填充；

S3-6、重复S3-3-S3-5，直至ROI集合中所有元素均被处理完成。

7.根据权利要求6所述的方法，其特征在于S3-2中，筛选公式为：

式中，sum_p,xel表示每个ROI内田埂像素占比，col_pixel表示纵向田埂在每个ROI内的占比，T₁和T₂分别表示成为待补全田埂ROI所需满足的条件阈值；i的取值由步长step和前一个ROI位置共同决定。

8.根据权利要求7所述的方法，其特征在于T₁＝1500，T₂＝100。

9.根据权利要求7所述的方法，其特征在于i＝ROI_before+7*step。

10.根据权利要求6所述的方法，其特征在于S3-4中，重新拟合的公式为：

式中，Road_area表示通过动态拟合后所得的最终待补全田埂区域，即新ROI，w′为依据像素拟合的新田埂宽，Half_L为ROI的半高，为ROI中点p_center的x坐标，θ为新的待补全田埂ROI与原ROI的夹角。