arXiv:2503.09187cs.CV2025-03被引 2

用YOLOv8检测边缘,精准生成高分辨率航拍图中的屋顶矢量图。

Polygonizing Roof Segments from High-Resolution Aerial Images Using Yolov8-Based Edge Detection

  • 以边缘检测为核心,结合几何关系重建屋顶结构。
  • 在SGA和Melville数据集上,矢量精度接近参考数据,mIoU达0.85–1,ovIoU近0.97。
  • 适用于复杂屋顶,对噪声和遮挡有强鲁棒性,适合城市建模应用。

本研究提出一种基于遥感图像的屋顶细节提取与矢量化新方法。不同于依赖角点检测的传统几何原型法,该方法以边缘检测为关键机制,通过几何关系推导角点与面。我们改进了原用于旋转目标检测的YOLOv8 OBB模型,有效提取屋顶边缘。在SGA与Melville数据集上的实验表明,该方法对噪声和遮挡具有强鲁棒性,可生成精确的屋顶矢量表示。在栅格层面,模型性能超越当前最优基础分割模型(SAM),多数样本mIoU在0.85至1之间,ovIoU接近0.97。在矢量层面,基于Hausdorff距离、PolyS指标及自定义栅格-矢量指标评估,多边形化后显著提升,逼近参考数据。该方法成功处理多样屋顶结构,修复边缘间隙,即使对训练集中未包含的新结构亦表现良好。研究证明其在自动屋顶矢量化中的潜力,支持城市地形重建等应用。

原文摘要 · Abstract (English)

This study presents a novel approach for roof detail extraction and vectorization using remote sensing images. Unlike previous geometric-primitive-based methods that rely on the detection of corners, our method focuses on edge detection as the primary mechanism for roof reconstruction, while utilizing geometric relationships to define corners and faces. We adapt the YOLOv8 OBB model, originally designed for rotated object detection, to extract roof edges effectively. Our method demonstrates robustness against noise and occlusion, leading to precise vectorized representations of building roofs. Experiments conducted on the SGA and Melville datasets highlight the method's effectiveness. At the raster level, our model outperforms the state-of-the-art foundation segmentation model (SAM), achieving a mIoU between 0.85 and 1 for most samples and an ovIoU close to 0.97. At the vector level, evaluation using the Hausdorff distance, PolyS metric, and our raster-vector-metric demonstrates significant improvements after polygonization, with a close approximation to the reference data. The method successfully handles diverse roof structures and refines edge gaps, even on complex roof structures of new, excluded from training datasets. Our findings underscore the potential of this approach to address challenges in automatic roof structure vectorization, supporting various applications such as urban terrain reconstruction.

屋顶矢量化边缘检测YOLOv8遥感

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