中国空间科学技术 ›› 2021, Vol. 41 ›› Issue (4): 134-141.doi: 10.16708/j.cnki.1000-758X.2021.0060

• 论文 • 上一篇    

一种深度学习分割遥感影像道路的损失函数

袁伟,许文波,周甜   

  1. 1 成都大学 建筑与土木工程学院,成都610106
    2 电子科技大学 资源与环境学院,成都610097
  • 出版日期:2021-08-25 发布日期:2021-07-30
  • 基金资助:
    四川省科技厅科技支撑计划(2020YFG0055)

A loss function of road segmentation in remote sensing image by deep learning

YUAN Wei,XU Wenbo,ZHOU Tian   

  1. 1 School of Architecture and Civil Engineering,Chengdu University,Chengdu 610106,China
    2 School of Resources and Environment,University of Electronic Science and Technology of China,Chengdu 610097,China
  • Online:2021-08-25 Published:2021-07-30

摘要: 传统的根据光谱特征或形态学算法来分割道路,存在精度低、阈值难确定等缺点,而深度学习中已有的方法并未考虑道路的特性,只是利用通用方法来分割道路。针对上述不足,提出了一种针对道路特有形态的深度学习损失函数——形态损失函数。首先使用连通性算法将预测结果划分为若干个相互分离的连通区域,分别计算这些区域的面积与外接圆面积的比值,然后取平均值作为此批训练数据的形态损失函数,最后将形态损失函数按一定的比例与交叉熵损失函数求和,得到最终的损失函数。通过在公开的遥感数据集上使用深度学习网络进行对比试验,附加了形态损失函数后平均交并比(MIoU)、准确度(ACC)及F1Score均有提高。从预测的图形来看,附加了形态损失函数后,预测的道路更为连续。所提出的形态损失函数可用于提高光学遥感影像道路分割的精度。

关键词: 形态学, 遥感影像, 道路分割, 语义分割, 深度学习

Abstract: Traditional road segmentation based on spectral features or morphological algorithms has some disadvantages such as low precision and difficulty in determining the threshold value, and the existing methods in deep learning do not consider the characteristics of roads, only using general methods to segment roads. A deep learning loss function named morphological loss function with road unique trait was proposed. Firstly, the connectivity algorithm was used to divide the prediction results into several separated connected regions, and the ratios of the region area to the circumscribed circle area was calculated respectively. Then, the average value of regions was taken as the morphological loss function of this batch of training data. Finally, the morphological loss function was summed with the cross entropy loss function according to a certain proportion to obtain the final loss function. Through the comparative experiment on open remote sensing dataset, MIoU, ACC and F1-Score were all improved by the addition of morphological loss function. According to the prediction image, the predicted road was more continuous when morphological loss function was added. The morphological loss function proposed is an effective method to improve the accuracy of road segmentation in remote sensing.

Key words: morphology, remote sensing image, road segmentation, semantic segmentation, deep learning