摘要
针对大部分国产卫星波段少,传统方法难以进行云检测。本文提出了基于改进DLinkNet模型的云检测算法,并应用于国产卫星遥感影像的云检测。首先使用人工勾云样本生成二值图标签;其次,对DLinkNet50的编码器进行改进,使用带有通道注意力机制的ResNeSt50-Block替换原始的ResNet50-Block;另外,对损失函数进行加权,测试分析发现,只用交叉熵损失作为损失函数,检测精度更高,最后使用条件随机场(CRF)对预测的结果进行后处理。实验结果表明,改进后的DLinkNet模型在训练集上的IoU提升2.2%,Precision提升1.2%,保持了较好的云边缘信息,可以很好地用于云检测,效果明显高于原DLinkNet模型。
-
单位安徽恒源煤电股份有限公司