作者投稿和查稿 主编审稿 专家审稿 编委审稿 远程编辑

计算机工程 ›› 2021, Vol. 47 ›› Issue (10): 306-313. doi: 10.19678/j.issn.1000-3428.0059312

• 开发研究与工程应用 • 上一篇    下一篇

基于深度学习的卫星图像道路分割算法

张新华1, 黄梦醒2, 张雨1, 李玉春2, 单怡晴1, 冯思玲2   

  1. 1. 海南大学 计算机与网络空间安全学院, 海口 570228;
    2. 海南大学 信息与通信工程学院, 海口 570228
  • 收稿日期:2020-08-20 修回日期:2020-10-27 发布日期:2021-10-11
  • 作者简介:张新华(1995-),女,硕士研究生,主研方向为计算机视觉、图像处理;黄梦醒(通信作者),教授、博士;张雨,副教授、博士;李玉春,博士研究生;单怡晴,硕士研究生;冯思玲,副教授、博士。
  • 基金资助:
    国家重点研发计划(2018YFB1703403,2018YFB1404400);海南省重点研发计划(ZDYF2019020);海南省高等学校科学研究项目(Hnky2019-22)。

Road Segmentation Algorithm for Satellite Images Based on Deep Learning

ZHANG Xinhua1, HUANG Mengxing2, ZHANG Yu1, LI Yuchun2, SHAN Yiqing1, FENG Siling2   

  1. 1. School of Computer Science and Cyberspace Security, Hainan University, Haikou 570228, China;
    2. School of Information and Communication Engineering, Hainan University, Haikou 570228, China
  • Received:2020-08-20 Revised:2020-10-27 Published:2021-10-11

摘要: 针对道路分割时存在的梯度消失问题,构建基于U-Net的卫星道路图像语义分割模型。通过密集连接模块减少梯度消失,并引入空间空洞金字塔结构保留更多的图像特征,在学习深层次特征信息时采用注意力监督机制,提取道路要素的特征信息。在卫星图像道路数据集上的测试结果表明,与FCN、SegNet、U_Net算法相比,该算法模型的准确率、召回率和精确率指标分别达到96.3%、96.9%和96.6%,能够有效地对道路元素进行准确分割。

关键词: 深度学习, 道路分割, 密集连接模块, 空间空洞金字塔结构, 注意力监督机制

Abstract: The existing road segmentation methods generally suffer from gradient disappearance, and are limited in feature utilization and semantic segmentation accuracy.To address the problem, a U-Net-based semantic segmentation model is proposed for satellite road images.The dense connection module is used to reduce gradient disappearance.Then the Atrous Spatial Pyramid Pooling(ASPP) structure is introduced to retain more image features.Finally, the attention monitoring mechanism is used when learning deep-level feature information, so as to more effectively extract the feature information of road elements.The test results on a road data set from satellite images show that compared with FCN, SegNet and U_Net algorithms, the proposed algorithm improves the accuracy to 96.3%, recall rate to 96.9% and precision to 96.6%.This algorithm can segment the road elements accurately.

Key words: deep learning, road segmentation, dense connection module, Atrous Spatial Pyramid Pooling(ASPP), attention monitoring mechanism

中图分类号: