中文 | English

J. Hefei Univ. Tech. (Nat. Sci.)

合肥工业大学学报(自然科学版)

JOURNAL OF HEFEI UNIVERSITY OF TECHNOLOGY (NATURAL SCIENCE)

Competent Authorities: Ministry of Education of the People's Republic of China

Sponsor by: Hefei University of Technology

ISSN 1000-1646, CN 34-1001/N

High-resolution remote sensing image building extraction based on UNet, UNet++, and Attention-UNet networks

Download Full Text

基于 UNet、UNet++、Attention-UNet 网络的遥感影像建筑物提取研究

Journal Information

J. Hefei Univ. Tech. (Nat. Sci.), Jun. 2026, Vol.49 No.6: 805-812

DOI: 10.3969/j.issn.1003-5060.2026.06.013

Authors

HUANG Miner, LI Zhenxuan, GAO Fei, TAO Tingye, ZHU Yongchao, YUAN Xiaoyu

(School of Civil and Hydraulic Engineering, Hefei University of Technology, Hefei 230009, China)

Abstract and Keywords

Abstract: This study selects three convolutional neural networks(CNNs), namely UNet, UNet++, and Attention-UNet, aiming to investigate the performance differences of this network series in building extraction. Based on the WHU building dataset, through experiments and quantitative evaluations, this research explores the effectiveness and applicability of the aforementioned network models in remote sensing image building extraction. The experimental results indicate that UNet++ achieves the highest extraction accuracy, followed by Attention-UNet and UNet, with accuracies of 0.987, 0.983, and 0.979, respectively. UNet++ exhibits superior edge extraction capabilities and correct building segmentation ratios compared to the other two networks. For small to medium-sized buildings, Attention-UNet demonstrates significant extraction effectiveness, but it is prone to false positives. UNet++ performs well with occasional instances of missed extractions, while the extraction effectiveness of UNet is subpar.

Keywords: building extraction; high-resolution remote sensing image; UNet; UNet++; Attention-UNet

作者信息

黄敏儿，李振轩，高飞，陶庭叶，朱勇超，袁啸宇

（合肥工业大学土木与水利工程学院，安徽合肥230009）

摘要和关键词

摘要: 文章选择 UNet、UNet++ 和 Attention-UNet 3 种卷积神经网络，研究该系列网络在建筑物提取中的性能差异。基于 WHU 建筑物数据集，通过实验和定量评估深入分析上述网络模型在遥感影像建筑物提取中的有效性和适用性。结果表明：UNet++ 的提取精度最高，其次是 Attention-UNet、UNet，分别为 0.987、0.983、0.979；UNet++ 的边缘提取能力和建筑物正确分割比例优于其他 2 种网络；针对中小型建筑物，Attention-UNet 的提取效果显著，但易出现非建筑物误提的情况，UNet++ 的提取效果良好，偶尔出现漏提现象，UNet 的提取效果欠佳。

关键词: 建筑物提取；高分辨率遥感影像；UNet 网络；UNet++网络；Attention-UNet 网络

Funding

国家自然科学基金资助项目（42104019）；安徽省自然科学基金资助项目（2208085QD105）和中央高校基本科研业务费专项资金资助项目（JZ2021HGTA0167）

Previous Next

Back to Contents