(19)国家知识产权局 (12)发明 专利申请 (10)申请公布号 (43)申请公布日 (21)申请 号 202210174578.3 (22)申请日 2022.02.24 (71)申请人 中国电建集团中南勘测设计 研究院 有限公司 地址 410014 湖南省长 沙市雨花区香樟东 路16号 (72)发明人 谭可成　刘昊　刘承照　许强红　 何维　马晨哲　胡文柯　 (74)专利代理 机构 长沙正奇专利事务所有限责 任公司 431 13 专利代理师 王娟 (51)Int.Cl. G06V 20/17(2022.01) G06V 10/26(2022.01) G06V 10/42(2022.01)G06V 10/774(2022.01) G06V 10/80(2022.01) G06V 10/82(2022.01) G06K 9/62(2022.01) G06N 3/04(2006.01) G06N 3/08(2006.01) (54)发明名称 道路分割方法及计算机装置 (57)摘要 本发明公开了一种道路分割方法及计算机 装置， 从无人机航拍高分辨率道路数据集获取N 张图像， 利用所述N张图像获取训练集； 利用所述 训练集训练道路提取模型， 得到训练后的道路提 取模型； 将待测试图像输入所述训练后的道路提 取模型， 得到道路提取结果。 本发明提出了双注 意力的残差学习模块， 在自适应调节感受野之外 兼顾了全局信息的提取， 能够更准确的捕捉道路 拓扑结构特征， 获得更拟合目标的感受野， 提升 了道路提取的连贯 性。 权利要求书2页 说明书8页 附图9页 CN 114550014 A 2022.05.27 CN 114550014 A 1.一种道路分割方法， 其特 征在于， 包括以下步骤： S1、 从无人机航拍高分辨 率道路数据集获取N张图像， 利用所述 N张图像获取训练集； S2、 利用所述训练集训练道路提取模型， 得到训练后的道路提取模型； 所述道路提取模型包括： 特征提取层， 用于对所述训练集中的图像进行降采样 操作； 至少一个双注意力残差模块， 包括至少一个残差单元， 所述残差单元用于对输入进行 包括如下步骤的操作： 对所述特 征提取层的输出进行 卷积操作， 得到特 征向量F1； 将所述特征向量F1输入通道注意力模块， 并将所述通道注意力模块的输出与所述特征 向量F1相乘， 得到第一输出 结果； 将所述第一输出结果输入空间注意力模块， 并将所述空间注意力模块的输出与所述第 一输出结果相乘， 得到第二输出 结果； 拼接所述特 征提取层的输出与所述第二输出 结果， 得到第三输出 结果； 上采样重建模块， 用于对所述第三结果进行上采样操作， 得到最终输出结果； S3、 将待 测试图像输入所述训练后的道路提取模型， 得到道路提取 结果。 2.根据权利要求1所述的道路分割方法， 其特征在于， 所述双注意力残差模块数量为两 个， 其中第一个双注意力残差模块的输出经池化层后输入第二个双注意力残差模块， 且两 个所述双注意力残差模块的输出拼接后的结果 为所述上采样重建模块的输入。 3.根据权利要求1所述的道路分割方法， 其特征在于， 所述双注意力残差模块数量为三 个， 其中第一个双注意力残差模块的输出经池化层后输入第二个双注意力残差模块， 第二 个双注意力残差模块的输出经池化层后输入第三个双注意力残差模块， 且三个所述双注意 力残差模块的输出拼接后的结果 为所述上采样重建模块的输入。 4.根据权利要求2或3所述的道路分割方法， 其特征在于， 每个所述双注意力残差模块 与一个全局金字塔聚合模块连接 。 5.根据权利要求4所述的道路分割方法， 其特征在于， 所有金字塔聚合模块的输出拼接 后得到的拼接结果输入所述上采样重 建模块； 或者， 第n个金字塔聚合模块的输出经第一上 采样层后与第n ‑1个金字塔聚合模块的输出拼接， 得到的第一拼接结果输入第二上采样层， 第二上采样层的输出与第n ‑2个金字塔采样模块的输出拼接后， 得到的第二拼接结果输入 第三上采样层， 依此类 推； 其中， n 为最后一个金字塔聚合模块， n≥3 。 6.根据权利要求1～5之一所述的道路分割方法， 其特征在于， 所述道路提取模型还包 括非局部注意力聚合模块， 所述 非局部注意力聚合模块的输入为所述双注意力残差模块的 输出经池化层操作后的结果； 所述非局部注意力聚合模块的输出与上一处理层的输出拼接 后输入所述上采样重 建模块； 其中， 所述上一处理层为最后一个双注意力残差模块， 或者最 后一个金字塔聚合模块。 7.根据权利要求6所述的道路分割方法， 其特征在于， 所述非局部注意力聚合模块与池 化层之间还设有可变形 卷积网络 。 8.根据权利要求1所述的道路分割方法， 其特征在于， 所述双注意力残差模块包括多个 串联的残差单元， 其中第一个残差单元 的输入为所述特征提取层的输出； 第二个残差单元 的输入为所述第一个残差单 元的输出， 依此类 推。权　利　要　求　书 1/2 页 2 CN 114550014 A 29.根据权利要求8所述的道路分割方法， 其特征在于， 当所述双注意力残差模块个数为 多个时， 每 个所述双注意力残差模块中残差单 元的数量 不同。 10.一种计算机装置， 包括存储器、 处理器及存储在存储器上的计算机程序； 其特征在 于， 所述处 理器执行所述计算机程序， 以实现权利要求1～ 9之一所述方法的步骤。权　利　要　求　书 2/2 页 3 CN 114550014 A 3