4 个月前

利用时空生成网络去除卫星图像中的云层

利用时空生成网络去除卫星图像中的云层

摘要

卫星图像在持续的环境监测和地球观测方面展现出巨大的潜力。然而,云层造成的遮挡会严重限制覆盖范围,使得地面信息提取变得更加困难。现有的处理流程通常通过简单的时序合成和手工设计的滤波器来去除云层。相比之下,我们将云层去除问题视为一个条件图像生成挑战,并提出了一种可训练的空间时间生成网络(Spatiotemporal Generator Network, STGAN)来去除云层。我们在构建的一个新的大规模时空数据集上训练了该模型,该数据集包含97,640对图像,涵盖了所有大陆。实验结果表明,所提出的STGAN模型优于标准模型,并且能够在各种大气条件下生成高PSNR和SSIM值的真实无云图像,从而在下游任务如土地覆盖分类中提高了性能。

代码仓库

come880412/CTGAN
pytorch
GitHub 中提及
PatrickTUM/SEN12MS-CR-TS
pytorch
GitHub 中提及
VSAnimator/stgan
pytorch
GitHub 中提及

基准测试

基准方法指标
cloud-removal-on-sen12ms-cr-tsSTGAN
PSNR: 25.42
RMSE: 0.057
SAM: 12.548
SSIM: 0.818

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
利用时空生成网络去除卫星图像中的云层 | 论文 | HyperAI超神经