CT-UNet: An Improved Neural Network Based on U-Net for Building Segmentation in Remote Sensing Images

التفاصيل البيبلوغرافية
العنوان: CT-UNet: An Improved Neural Network Based on U-Net for Building Segmentation in Remote Sensing Images
المؤلفون: Haohao Cheng, Huanran Ye, Sheng Liu, Kun Jin
المصدر: ICPR
بيانات النشر: IEEE, 2021.
سنة النشر: 2021
مصطلحات موضوعية: Contextual image classification, Channel (digital image), Computer science, Feature extraction, 0211 other engineering and technologies, Context (language use), 02 engineering and technology, Image segmentation, 0202 electrical engineering, electronic engineering, information engineering, 020201 artificial intelligence & image processing, Segmentation, Spatial analysis, 021101 geological & geomatics engineering, Remote sensing, Block (data storage)
الوصف: With the proliferation of remote sensing images, how to segment buildings more accurately in remote sensing images is a critical challenge. First, the high resolution leads to blurred boundaries in the extracted building maps. Second, the similarity between buildings and background results in intra-class inconsistency. To address these two problems, we propose an UNet-based network named Context-Transfer-UNet (CT-UNet). Specifically, we design Dense Boundary Block (DBB). Dense Block utilizes reuse mechanism to refine features and increase recognition capabilities. Boundary Block introduces the low-level spatial information to solve the fuzzy boundary problem. Then, to handle intra-class inconsistency, we construct Spatial Channel Attention Block (SCAB). It combines context space information and selects more distinguishable features from space and channel. Finally, we propose a novel loss function to enhance the purpose of loss by adding evaluation indicator. Based on our proposed CT-UNet, we achieve 85.33% mean IoU on the Inria dataset and 91.00% mean IoU on the WHU dataset, which outperforms our baseline (U-Net ResNet-34) by 3.76% and Web-Net by 2.24%.
URL الوصول: https://explore.openaire.eu/search/publication?articleId=doi_________::f96c9169e417949e445958c2f6f397b7
https://doi.org/10.1109/icpr48806.2021.9412355
حقوق: CLOSED
رقم الأكسشن: edsair.doi...........f96c9169e417949e445958c2f6f397b7
قاعدة البيانات: OpenAIRE