Accurate measurement of the offset from roof-to-footprint in very-high-resolution remote sensing imagery is crucial for urban information extraction tasks. With the help of deep learning, existing methods typically rely on two-stage CNN models to extract regions of interest on building feature maps. At the first stage, a Region Proposal Network (RPN) is applied to extract thousands of ROIs (Region of Interests) which will post-imported into a Region-based Convolutional Neural Networks (RCNN) to extract wanted information. However, because of inflexible RPN, these methods often lack effective user interaction, encounter difficulties in instance correspondence, and struggle to keep up with the advancements in general artificial intelligence. This paper introduces an interactive Transformer model combined with a prompt encoder to precisely extract building segmentation as well as the offset vectors from roofs to footprints. In our model, a powerful module, namely ROAM, was tailored for common problems in predicting roof-to-footprint offsets. We tested our model's feasibility on the publicly available BONAI dataset, achieving a significant reduction in Prompt-Instance-Level offset errors ranging from 14.6% to 16.3%. Additionally, we developed a Distance-NMS algorithm tailored for large-scale building offsets, significantly enhancing the accuracy of predicted building offset angles and lengths in a straightforward and efficient manner. To further validate the model's robustness, we created a new test set using 0.5m remote sensing imagery from Huizhou, China, for inference testing. Our code, training methods, and the updated dataset will be accessable at https://github.com/likaiucas.
翻译:在高分辨率遥感影像中精确测量屋顶到基底偏移量,对于城市信息提取任务至关重要。借助深度学习技术,现有方法通常依赖两阶段CNN模型来提取建筑特征图上的感兴趣区域。第一阶段采用区域建议网络提取数千个感兴趣区域,随后输入基于区域的卷积神经网络提取所需信息。然而,由于区域建议网络缺乏灵活性,这些方法难以实现有效的用户交互,在实例对应方面存在困难,且难以跟上通用人工智能的发展步伐。本文提出一种结合提示编码器的交互式Transformer模型,用于精确提取建筑分割及其屋顶到基底的偏移向量。针对屋顶-基底偏移预测中的常见问题,我们专门设计了名为ROAM的强力模块。在公开的BONAI数据集上验证了模型可行性,提示实例级偏移误差显著降低14.6%-16.3%。此外,我们针对大规模建筑偏移开发了Distance-NMS算法,以简洁高效的方式显著提升了建筑偏移角度和长度的预测精度。为验证模型鲁棒性,我们采用中国惠州0.5米分辨率遥感影像构建新测试集进行推理测试。相关代码、训练方法及更新后的数据集已在https://github.com/likaiucas 公开。