测绘通报 ›› 2026, Vol. 0 ›› Issue (7): 136-141.doi: 10.13474/j.cnki.11-2246.2026.0720

• 技术交流 • 上一篇    下一篇

融合YOLO与CBAM模型的建筑物轮廓遥感提取方法

毛亚琴1,2, 丁小辉3, 涂梨平1, 范军林1,2, 何雨玥3, 徐荣华4   

  1. 1. 江西省核工业地质调查院, 江西 南昌 330038;
    2. 江西核工业测绘院集团有限公司, 江西 南昌 330038;
    3. 江西财经大学, 江西 南昌 330013;
    4. 江西省地质调查勘查院关键矿产资源勘查与开发江西省重点实验室, 江西 南昌 330009
  • 收稿日期:2025-11-05 发布日期:2026-08-15
  • 作者简介:毛亚琴(1992—),女,主要研究方向为遥感测绘、三维建模。E-mail:lemonmyq0805@163.com
  • 基金资助:
    国家自然科学基金(42201467);江西省地质局青年科学技术带头人培养计划(2025JXDZKJRC06)

A remote sensing extraction method for building contours based on YOLO and CBAM model fusion

Mao Yaqin1,2, Ding Xiaohui3, Tu Liping1, Fan Junlin1,2, He Yuyue3, Xu Ronghua4   

  1. 1. Jiangxi Provincial Institute of Nuclear Geological Survey, Nanchang 330038, China;
    2. Jiangxi Nuclear Industry Surveying and Mapping Institute Group Co., Ltd., Nanchang 330038, China;
    3. Jiangxi University of Finance and Economics, Nanchang 330013, China;
    4. Jiangxi Geological Survey and Exploration Institute(Jiangxi Province Key Laboratory of Exploration and Development of Critical Mineral Resources), Nanchang 330009, China
  • Received:2025-11-05 Published:2026-08-15

摘要: [目的] 建筑物轮廓遥感提取是自然资源调查与空间规划的重要基础。针对基于遥感的建筑物轮廓提取面临的因建筑物轮廓几何特征复杂而引起的提取精度低等问题,本文提出了融合YOLO11模型与卷积块注意力模块(CBAM)的YOLO-CBAM模型,用于建筑物轮廓遥感提取。[方法] 模型采用公开的GF-7建筑物数据集进行训练与测试,并与YOLOv8、YOLO11、融合YOLO11与CA(coordinate attention)注意力机制的YOLO-CA模型、U-Net、DeepLabV3及Mask-RCNN模型进行了比较。[结果] 试验结果显示,YOLO-CBAM模型的IoU、Dice指数、mAP@0.5值、F1值及最大召回率(R)分别为71.1%、81.5%、74.7%、0.72及88.0%。[结论] 这些评价指标的值均优于YOLOv8、YOLO11及YOLO-CA模型,表明CBAM的引入可以有效提升YOLO11模型对遥感建筑物轮廓提取的精度。

关键词: YOLO模型, 建筑物轮廓, 高分遥感, 注意力机制

Abstract: [Purposes] Remote sensing extraction of building outlines serves as a vital source of foundational data for natural resource surveys and spatial planning.Addressing the challenge of low extraction accuracy,stemming from the complex geometric characteristics of building contours in remote sensing imagery,this study proposes the YOLO-CBAM model,which integrates the YOLO11 framework with the convolutional block attention module (CBAM)for enhanced remote sensing building contour extraction. [Methods] The model is trained and validated using the public GF-7 building dataset and compared against YOLOv8,YOLO11,the YOLO-CA model (incorporating the Coordinate Attention mechanism with YOLO11),U-Net,DeepLabV3,and Mask-RCNN. [Findings] Experimental results demonstrate that the YOLO-CBAM model achieves IoU,Dice index,mAP@0.5,F1 score,maximum recall(R) values of 71.1%,81.5%,74.7%,0.72,88.0%,respectively. [Conclusions] These evaluation metrics outperform those of YOLOv8,YOLO11,YOLO-CA,U-Net,DeepLabV3,and Mask-RCNN confirming that the integration of CBAM significantly improves the accuracy of YOLO11 in remote sensing-based building contour extraction.

Key words: YOLO model, building contours, high-resolution remote sensing, attention mechanism

中图分类号: