arXiv:2409.04025cs.CVcs.AI2024-09被引 27

针对建筑立面小物体检测难问题,提出BFA-YOLO模型提升精度。

BFA-YOLO: A balanced multiscale object detection network for building façade attachments detection

论文配图:BFA-YOLO: A balanced multiscale object detection network for building façade attachments detection
图 1 · 摘自论文原文
  • 设计特征平衡纺锤模块解决物体分布不均问题
  • 在自建数据集上比YOLOv8提升2.9% mAP50
  • 适合智能BIM与多视角立面检测场景

建筑立面元素(如门窗、空调外机、广告牌)的检测是自动化建筑信息模型(BIM)构建的关键步骤。然而,该任务面临元素分布不均、小物体多及背景噪声大等挑战。本文提出BFA-YOLO模型和BFA-3D数据集以应对这些问题。BFA-YOLO引入三个新组件:特征平衡纺锤模块(FBSM)缓解物体分布不均;目标动态对齐检测头(TDATH)增强小物体检测;位置记忆自注意力机制(PMESA)降低背景干扰。所构建的BFA-3D数据集包含多视角图像与精细标注,覆盖广泛立面元素类别。实验表明,BFA-YOLO在BFA-3D数据集上相比YOLOv8提升1.8% mAP$_{50}$,在公开的Façade-WHU数据集上提升2.9%。结果验证了模型在立面元素检测中的优越性,推动了智能BIM技术发展。

原文摘要 · Abstract (English)

The detection of façade elements on buildings, such as doors, windows, balconies, air conditioning units, billboards, and glass curtain walls, is a critical step in automating the creation of Building Information Modeling (BIM). Yet, this field faces significant challenges, including the uneven distribution of façade elements, the presence of small objects, and substantial background noise, which hamper detection accuracy. To address these issues, we develop the BFA-YOLO model and the BFA-3D dataset in this study. The BFA-YOLO model is an advanced architecture designed specifically for analyzing multi-view images of façade attachments. It integrates three novel components: the Feature Balanced Spindle Module (FBSM) that tackles the issue of uneven object distribution; the Target Dynamic Alignment Task Detection Head (TDATH) that enhances the detection of small objects; and the Position Memory Enhanced Self-Attention Mechanism (PMESA), aimed at reducing the impact of background noise. These elements collectively enable BFA-YOLO to effectively address each challenge, thereby improving model robustness and detection precision. The BFA-3D dataset, offers multi-view images with precise annotations across a wide range of façade attachment categories. This dataset is developed to address the limitations present in existing façade detection datasets, which often feature a single perspective and insufficient category coverage. Through comparative analysis, BFA-YOLO demonstrated improvements of 1.8\% and 2.9\% in mAP$_{50}$ on the BFA-3D dataset and the public Façade-WHU dataset, respectively, when compared to the baseline YOLOv8 model. These results highlight the superior performance of BFA-YOLO in façade element detection and the advancement of intelligent BIM technologies.

目标检测建筑信息模型小物体检测

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。