改进D-LinkNet模型在国产卫星影像云检测中的应用

作者：刘广进; 王光辉^*; 毕卫华; 刘慧杰; 杨化超

来源：测绘通报, 2021, (11): 54-64.

DOI：10.13474/j.cnki.11-2246.2021.338

• 云检测
• D-LinkNet
• 注意力机制
• 国产卫星
• 条件随机场

大部分国产卫星波段少,利用传统方法难以进行云检测。本文提出了基于改进D-LinkNet模型的云检测算法,并应用于国产卫星遥感影像的云检测。首先,利用自然资源部国土卫星遥感应用中心提供的人工勾云样本生成二值图标签;其次,对D-LinkNet50的编码器进行改进,使用带有通道注意力机制的ResNeSt50-Block替换原始的ResNet50-Block;然后,对损失函数进行加权,测试分析发现只用交叉熵损失作为损失函数时,检测精度更高;最后,使用条件随机场(CRF)对预测结果进行后处理。试验结果表明,改进D-LinkNet模型在测试集上的IoU提升了1.93%,精度提升了2.45%,保持了较好的云边缘信息,云检测效果高于原D-LinkNet模型。

• 单位
  中国矿业大学; 中国矿业大学（北京）