融合域自适应和弱监督策略的遥感影像建筑物提取算法

摘要

常见的建筑物提取算法主要采用全监督的方式实现,得到的模型通常在训练数据集上表现良好,而在跨区域使用时效果不佳。基于生成对抗网络的域自适应方法虽然在一定程度上能够增强网络的迁移能力,但由于缺乏目标域关键信息,效果难以保证。设计了一种全新的端到端弱监督建筑物提取网络。首先,采用像素关联模块来提升生成网络的性能;然后,在此基础上综合运用域自适应和图像级弱标签两种策略来优化训练过程,从而大幅提升了网络模型的泛化扩展性能。采用3组数据对所提方法的有效性进行了验证,通过大量实验证明了所提方法可以有效提升建筑物提取的性能。同时,通过消融实验验证了网络中各个模块的有效性。