ISSN 1009-5624 CN 10-2021/TQ 主管:中国乐凯集团有限公司 主办:北京乐凯科技有限公司
【摘要】云和阴影的检测与分割在遥感图像处理中具有重要意义。 本研究通过对比三种不同的深度学习模型结构:U 形网络(U⁃shaped network,UNet)、改进的 UNet++以及结合了 Swin Transformer(分层视觉 Transformer,使用偏移窗口)的 Swin⁃UNet(分割改编),评估它们在小数据集上对云和阴影分割效果。 实验数据集选用遥感影像去云数据集,包含 450 组带有 label(标签)和 mask(掩模)的遥感图像。 实验结果表明,尽管 Swin⁃UNet 模型在准确率上略高,达到了 81???? 39%,但在小数据集上易出现过拟合和训练波动问题;改进后的 UNet++模型增加了多个跳跃连接和批归一化层,最高准确率达到 80???? 18%,且训练过程中的波动较小。 UNet++模型在保持高准确率的同时,显著提高了模型的稳定性和泛化能力。 本文的研究为在资源有限环境下进行高效的云和阴影分割提供了理论依据和实践指导,证明了适当改进的小模型在小数据集上的应用潜力。