基于注意力机制和特征聚合的航拍输电线分割方法

本发明公开了基于注意力机制和特征聚合的航拍输电线分割方法，包括S1，构建输电线分割数据集，进行数据增强操作，并划分训练集、验证集和测试集；S2，将数据集中的图像输入神经网络模型进行训练，使用加权混合损失函数约束网络收敛，S2包括以下步骤：S21，在编码器和解码器跳跃连接的部分添加融合注意力模块；S22，设计特征聚合模块，添加在编码器第五层的后面；S23，将上述模块添加进U‑Net网络，构建神经网络模型并训练网络模型，使用加权混合损失函数指导网络进行训练。本发明通过添加融合注意力模块，增强网络对全局上下文信息的感知能力，同时提出了特征聚合模块来获取更加高级的语义特征，有效提高了输电线分割的准确性。