双通道对抗网络在建筑物分割中的应用

• 全文链接
• 请求原文

作   者：

张文政;  吴长悦;  满卫东;  刘明月;  赵文; 

作者机构：

华北理工大学矿业工程学院; 

关键词：

小波变换;  建筑物分割;  无人机;  对抗网络;  遮挡恢复; 

期刊名称：

计算机工程

i s s n：

1000-3428

年卷期：

2024 年 50 卷 011 期

页   码：

297-307

摘   要：

无人机影像中的建筑物提取主要面临两大挑战，首先是容易受到树木和阴影的遮挡，导致分割错误，其次是现有方法常忽略建筑物的形态和多分辨率信息。为了应对这些挑战，在对抗网络的框架内引入双通道并行的生成器策略，其中一个通道基于形态驱动的小波变换，专注于捕获建筑物的形态属性，包括建筑物的轮廓和结构特征，另一个通道基于DeepLabv3+,用于处理建筑物的复杂纹理，包括表面纹理和细节信息，这种设计使网络可以从多个方面理解影像中的建筑物特征。同时，为了应对遮挡问题，提出一种遮挡感知预处理模块，该模块能够有效地从深度信息中还原被树木和阴影遮挡的建筑轮廓和纹理信息。为了进一步提高网络识别建筑物特征的能力，通过特征融合模块引入自适应注意力机制，并实现一个复合损失函数来增强模型对建筑物结构和形态的敏感度。在两个不同的建筑物数据集上进行实验，结果表明，该网络的平均交并比(mIoU)分别达到93.60%和96.60%,F1分数、准确率也分别达到94.90%、94.42%和95.90%、96.42%。实验数据显示，所提网络可以恢复被遮挡信息同时提高分割精度，为城市规划、资源管理等应用提供有力支持。