arXiv:2411.16733cs.CV2024-11CVPR被引 24

构建全球尺度道路图数据集并提出新模型,提升遥感图像道路提取精度。

Towards Satellite Image Road Graph Extraction: A Global-Scale Dataset and A Novel Method

  • 采用节点引导重采样方法改进现有模型训练与推理不一致问题。
  • 数据集覆盖超1.38万平方公里,规模达现有最大数据集20倍。
  • 适用于遥感、自动驾驶等需要高精度道路信息的场景。

道路图提取在自动驾驶、导航等领域具有关键作用,但因标注数据严重匮乏,准确高效地提取道路图仍是难题。为此,我们构建了一个全球尺度的卫星图像道路图提取数据集——Global-Scale数据集,其规模约为现有最大公开数据集的20倍,覆盖超过13,800 km²。同时,提出新型道路图提取模型SAM-Road++,采用节点引导重采样方法缓解SAM-Road模型在训练与推理阶段的不匹配问题,并引入简单有效的“扩展线”策略以减轻道路遮挡影响。大量实验验证了该数据集与SAM-Road++方法的有效性,尤其在未见区域表现出优异的预测能力。相关数据集与代码已开源。

原文摘要 · Abstract (English)

Recently, road graph extraction has garnered increasing attention due to its crucial role in autonomous driving, navigation, etc. However, accurately and efficiently extracting road graphs remains a persistent challenge, primarily due to the severe scarcity of labeled data. To address this limitation, we collect a global-scale satellite road graph extraction dataset, i.e. Global-Scale dataset. Specifically, the Global-Scale dataset is $\sim20 \times$ larger than the largest existing public road extraction dataset and spans over 13,800 $km^2$ globally. Additionally, we develop a novel road graph extraction model, i.e. SAM-Road++, which adopts a node-guided resampling method to alleviate the mismatch issue between training and inference in SAM-Road, a pioneering state-of-the-art road graph extraction model. Furthermore, we propose a simple yet effective ``extended-line'' strategy in SAM-Road++ to mitigate the occlusion issue on the road. Extensive experiments demonstrate the validity of the collected Global-Scale dataset and the proposed SAM-Road++ method, particularly highlighting its superior predictive power in unseen regions. The dataset and code are available at \url{https://github.com/earth-insights/samroadplus}.

遥感图像道路提取数据集深度学习

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