一种基于Swift-SegEdgeNet的遥感图像道路提取任务的实现方法

本发明公开了一种基于Swift‑SegEdgeNet的遥感图像道路提取任务的实现方法，包括以下步骤：首先，对Massachusetts道路数据集和DeepGlobe 2018道路数据集进行预处理并制作边缘标签；其次，构建Swift‑SegEdgeNet模型；随后，将预处理后并制作好边缘标签的Massachusetts道路数据集和DeepGlobe 2018道路数据集放入Swift‑SegEdgeNet中进行训练；最后，使用mIOU、mDC、Recall以及F1‑score对道路提取结果进行评价。本发明利用多任务思想对网络结构进行优化并且提出双分支卷积模块提取道路方向特征，在遥感图像道路提取任务中，分割准确率超过FCN、UNet、SegNet、ResUNet、AttUNet、DUNet、LinkNet34、DinkNet34、UNet++、UNet2++、TransUNet、Cswin、Segformer、Swin‑ResUNet、Swin‑ResUNet+等网络。这对遥感图像自动化提取道路具有较大的实际意义。