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

通过几何感知半监督学习从街景图像中进行细粒度建筑功能识别

Fine-Grained Building Function Recognition from Street-View Images via Geometry-Aware Semi-Supervised Learning

School of Geospatial Engineering and Science, Sun Yat-sen University(中山大学地理空间工程与科学学院);Tsinghua University(清华大学);Shanghai Artificial Intelligence Laboratory(上海人工智能实验室);SenseTime Research(商汤科技研究院);Tsinghua Shenzhen International Graduate School, Tsinghua University(清华大学深圳国际研究生院)

Weijia Li, Jinhua Yu, Dairong Chen, Yi Lin, Runmin Dong, Xiang Zhang, Conghui He, Haohuan Fu

AI总结:

本文提出几何感知半监督框架,融合 GIS 与街景几何关系生成粗伪标注,实现跨城市细粒度建筑功能识别,性能显著优于全监督与现有半监督方法。

摘要:

在本研究中,我们提出了一种用于细粒度建筑功能识别的几何感知半监督框架,利用多源数据之间的几何关系来提高半监督学习中伪标签的准确性,并拓宽其对各类建筑功能分类体系的适用性。首先,我们设计了一个在线半监督预训练阶段,这有助于精确获取街景图像中建筑立面的位置信息。在第二阶段,我们提出了一个几何感知的粗标注生成模块。该模块基于几何关系有效结合 GIS 数据和街景数据,提高了伪标注的准确性。在第三阶段,我们将新生成的粗标注与现有标注数据集相结合,以在多个城市大规模实现建筑的细粒度功能识别。大量实验表明,我们提出的框架在建筑细粒度功能识别方面表现出优越性能。在同一分类体系内,与全监督方法和最先进的半监督方法相比,它分别实现了 7.6% 和 4.8% 的提升。此外,我们的方法在跨城市场景中也表现良好,即将在 OmniCity(纽约)上训练的模型扩展到具有不同建筑功能分类体系的新城市(即洛杉矶和波士顿)。本研究为以最低标注需求开展大规模多城市应用提供了一种新的解决方案,有助于城市管理中更高效的数据更新和资源配置。

英文摘要:

In this work, we propose a geometry-aware semi-supervised framework for fine-grained building function recognition, utilizing geometric relationships among multi-source data to enhance pseudo-label accuracy in semi-supervised learning, broadening its applicability to various building function categorization systems. Firstly, we design an online semi-supervised pre-training stage, which facilitates the precise acquisition of building facade location information in street-view images. In the second stage, we propose a geometry-aware coarse annotation generation module. This module effectively combines GIS data and street-view data based on the geometric relationships, improving the accuracy of pseudo annotations. In the third stage, we combine the newly generated coarse annotations with the existing labeled dataset to achieve fine-grained functional recognition of buildings across multiple cities at a large scale. Extensive experiments demonstrate that our proposed framework exhibits superior performance in fine-grained functional recognition of buildings. Within the same categorization system, it achieves improvements of 7.6\% and 4.8\% compared to fully-supervised methods and state-of-the-art semi-supervised methods, respectively. Additionally, our method also performs well in cross-city scenarios, i.e., extending the model trained on OmniCity (New York) to new cities (i.e., Los Angeles and Boston) with different building function categorization systems. This study offers a new solution for large-scale multi-city applications with minimal annotation requirements, facilitating more efficient data updates and resource allocation in urban management.

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

↑