公开(公告)号：CN118506204A

公开(公告)日：2024-08-16

申请号：CN202410689224.1

申请日：2024-05-30

Applicant: 武汉大学 ,  湖北珞珈实验室

Inventor： 陈渠森 ,  刘瑞淇 ,  姜卫平 ,  宋成芳 ,  李昭

IPC: G06V20/13 ,  G06V20/10 ,  G06V10/82 ,  G06N3/0455 ,  G06N3/0464 ,  G06N3/042 ,  G06V10/764 ,  G06N3/0475 ,  G06N3/094 ,  G06V10/74 ,  G06N3/048 ,  G06N3/047

Abstract: 本发明公开了一种融合像素增强和图推理的遥感影像道路提取方法，包括S1，基于DeepGlobe制作预处理的遥感影像道路提取数据集；S2，预增强训练数据；S3，搭建模型架构；S4，损失函数设计；S5，在数据集上对上述模型进行训练；S6，在测试集上对训练好的模型进行评估。本发明的有益效果如下：(1)该方法采用基于Transformer的道路像素增强模块，以预增强的道路像素的形式保留了不同区域道路像素之间的远距离依赖关系，提高了后续道路分割的像素级准确度。(2)基于图卷积推理的分割模块纳入像素分类和方向预测分支，依靠节点关联性的推理实现道路提取，解决了传统的语义分割方法提取道路不连续的问题。