一种基于混合注意力机制与方向先验的道路提取方法

本发明涉及一种基于混合注意力机制与方向先验的道路提取方法，包括以下步骤：获取遥感影像，输入训练好的基于混合注意力机制与方向先验的道路提取模型，获得道路提取结果；上述模型包括编码器和解码器，其中编码器包括多个依次相连的混合注意力模块，每个混合注意力模块中，输入数据首先通过多个依次相连的稀疏注意力层提取局部特征，最后通过一个改进密集注意力层捕获全局语义关联；解码器包括方向增强融合模块和多个轻量混合注意力模块，所述方向增强融合模块在多个方向上进行卷积操作，并通过动态调整获得融合特征图。与现有技术相比，本发明可以在提高计算效率的同时，确保道路提取结果的连续性和精确性。
同济大学
刘世杰 | 贺钒 | 童小华 | 柳思聪 | 谢欢 | 冯永玖 | 金雁敏