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

从大幅面甚高分辨率遥感影像中学习整体检测桥梁

Learning to Holistically Detect Bridges from Large-Size VHR Remote Sensing Imagery

Wuhan University(武汉大学);Huazhong University of Science and Technology(华中科技大学);South China University of Technology(华南理工大学);ByteDance AI Lab(字节跳动人工智能实验室)

Yansheng Li, Junwei Luo, Yongjun Zhang, Yihua Tan, Jin-Gang Yu, Song Bai

AI总结:

本文针对大幅面甚高分辨率遥感影像中桥梁整体检测难题,提出包含6000幅影像和59737座桥梁的GLH-Bridge数据集及HBD-Net网络,通过SDFF架构与SSRW策略提升检测性能,并验证了其有效性与泛化能力。

摘要:

遥感影像中的桥梁检测在各种应用中发挥着至关重要的作用,但与其他物体的检测相比,它带来了独特的挑战。在遥感影像中,桥梁在空间尺度和长宽比方面表现出相当大的变化。因此,为了确保桥梁的可见性和完整性,在大幅面甚高分辨率遥感影像中进行整体桥梁检测至关重要。然而,缺乏大幅面甚高分辨率遥感影像数据集限制了深度学习算法在桥梁检测上的性能。由于GPU内存在处理大幅面图像时的限制,基于深度学习的物体检测方法通常采用裁剪策略,这不可避免地导致标签碎片化和不连续的预测。为了缓解数据集的稀缺性,本文提出了一个名为GLH-Bridge的大规模数据集,包含从全球不同地理位置采样的6,000幅甚高分辨率遥感影像。这些图像涵盖广泛的尺寸范围,从2,048×2,048到16,384×16,384像素不等,并总共包含59,737座桥梁。此外,我们提出了一种用于大幅面遥感影像中整体桥梁检测的高效网络(HBD-Net)。HBD-Net提出了一种基于分离检测器的特征融合(SDFF)架构,并通过形状敏感样本重加权(SSRW)策略进行优化。基于所提出的GLH-Bridge数据集,我们建立了一个包括OBB和HBB任务的桥梁检测基准,并验证了所提出的HBD-Net的有效性。此外,在两个公开可用数据集上的跨数据集泛化实验表明了GLH-Bridge数据集强大的泛化能力。

英文摘要:

Bridge detection in remote sensing images (RSIs) plays a crucial role in various applications, but it poses unique challenges compared to the detection of other objects. In RSIs, bridges exhibit considerable variations in terms of their spatial scales and aspect ratios. Therefore, to ensure the visibility and integrity of bridges, it is essential to perform holistic bridge detection in large-size very-high-resolution (VHR) RSIs. However, the lack of datasets with large-size VHR RSIs limits the deep learning algorithms' performance on bridge detection. Due to the limitation of GPU memory in tackling large-size images, deep learning-based object detection methods commonly adopt the cropping strategy, which inevitably results in label fragmentation and discontinuous prediction. To ameliorate the scarcity of datasets, this paper proposes a large-scale dataset named GLH-Bridge comprising 6,000 VHR RSIs sampled from diverse geographic locations across the globe. These images encompass a wide range of sizes, varying from 2,048*2,048 to 16,38*16,384 pixels, and collectively feature 59,737 bridges. Furthermore, we present an efficient network for holistic bridge detection (HBD-Net) in large-size RSIs. The HBD-Net presents a separate detector-based feature fusion (SDFF) architecture and is optimized via a shape-sensitive sample re-weighting (SSRW) strategy. Based on the proposed GLH-Bridge dataset, we establish a bridge detection benchmark including the OBB and HBB tasks, and validate the effectiveness of the proposed HBD-Net. Additionally, cross-dataset generalization experiments on two publicly available datasets illustrate the strong generalization capability of the GLH-Bridge dataset.

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

↑