arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2503.09187cs.CV

基于Yolov8边缘检测的高分辨率航空影像屋顶分段多边形化方法

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

Technical University of Darmstadt(达姆施塔特工业大学);Fraunhofer IOSB Ettlingen(弗劳恩霍夫应用研究促进协会IOSB埃特林根分部)

Qipeng Mei, Dimitri Bulatov, Dorota Iwaszczuk

AI总结:

本研究提出以YOLOv8 OBB边缘检测为核心、结合几何关系定义角点与面的屋顶矢量化方法,在SGA和Melville数据集上栅格与矢量层面均显著优于SAM等现有方法,能鲁棒处理复杂屋顶结构并支持城市地形重建。

摘要:

本研究提出了一种利用遥感影像进行屋顶细节提取与矢量化的新方法。与以往依赖角点检测的基于几何基元的方法不同,本方法以边缘检测作为屋顶重建的主要机制,同时利用几何关系来定义角点和面。我们改进了原本用于旋转目标检测的YOLOv8 OBB模型,以有效提取屋顶边缘。本方法对噪声和遮挡表现出鲁棒性,能够生成精确的建筑屋顶矢量化表示。在SGA和Melville数据集上进行的实验验证了该方法的有效性。在栅格层面,我们的模型优于最先进的基础分割模型(SAM),大多数样本的mIoU在0.85到1之间,ovIoU接近0.97。在矢量层面,使用Hausdorff距离、PolyS指标以及我们的栅格-矢量指标进行评估,结果表明多边形化后有了显著改进,与参考数据高度接近。该方法能够成功处理多样的屋顶结构,并修复边缘缺口,即使在训练数据集中未包含的复杂屋顶结构上也能有效工作。我们的研究结果凸显了该方法在应对自动屋顶结构矢量化挑战方面的潜力,可支持城市地形重建等多种应用。

英文摘要:

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.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑