基于SETR的自然保护区遥感影像输电线路精细提取方法

本发明公开了一种基于SETR的自然保护区遥感影像输电线路精细提取方法，涉及图像识别技术领域，包括构造数据集；用SETR网络对遥感图片进行类别预测，得到预测标签图；定义数个一阶卷积核、二阶卷积核及对应的计算公式、边缘强度的计算公式；构造边缘对齐损失、曲率损失、第t轮次训练的总损失函数，设置训练策略并训练SETR网络。本发明根据输电线路的使用场景和形状特点，结合边缘对齐损失、曲率损失平滑轮廓，能准确将输电线路与背景区分开、使用一阶、二阶梯度卷积核实现对于斜向或曲线边缘的检测，并使用动态调整损失权重策略，有效地平衡模型对基础任务和细节优化的学习需求，最终提高输电线路的识别准确度和可靠性。