arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2408.07613cs.CV

再思考遥感立体匹配网络泛化的关键因素

Rethinking the Key Factors for the Generalization of Remote Sensing Stereo Matching Networks

Aerospace Information Research Institute, Chinese Academy of Sciences(中国科学院空天信息创新研究院);School of Electronic, Electrical and Communication Engineering, University of Chinese Academy of Sciences(中国科学院大学电子电气与通信工程学院)

Liting Jiang, Feng Wang, Wenyi Zhang, Peifeng Li, Hongjian You, Yuming Xiang

AI总结:

针对遥感立体匹配监督数据不足、跨域泛化难的问题,从数据集选择、模型结构、训练方式三方面研究关键因素,提出泛化性好的无监督立体匹配网络并开源代码数据。

摘要:

立体匹配是三维重建的关键步骤,凭借其对遥感图像强大的特征表示能力,已完全转向深度学习方法。然而,立体匹配任务的真实标签依赖昂贵的机载LiDAR数据,因此难以获取足够样本用于监督学习。为提升立体匹配网络在来自不同传感器和场景的跨域数据上的泛化能力,本文从三个角度研究关键训练因素。(1)在训练数据集选择方面,选择与测试集区域目标分布相似的数据比使用同传感器数据更重要。(2)在模型结构方面,优先选择可灵活适配不同尺寸特征的级联结构。(3)在训练方式方面,无监督方法比监督方法泛化性更好,我们设计了一种无监督早停策略,以预训练权重为基础帮助保留最优模型。我们开展了大量实验以支撑上述发现,在此基础上提出了一种具有良好泛化性能的无监督立体匹配网络。我们在https://github.com/Elenairene/RKF_RSSM 发布了源代码和数据集,用于复现结果并推动后续研究。

英文摘要:

Stereo matching, a critical step of 3D reconstruction, has fully shifted towards deep learning due to its strong feature representation of remote sensing images. However, ground truth for stereo matching task relies on expensive airborne LiDAR data, thus making it difficult to obtain enough samples for supervised learning. To improve the generalization ability of stereo matching networks on cross-domain data from different sensors and scenarios, in this paper, we dedicate to study key training factors from three perspectives. (1) For the selection of training dataset, it is important to select data with similar regional target distribution as the test set instead of utilizing data from the same sensor. (2) For model structure, cascaded structure that flexibly adapts to different sizes of features is preferred. (3) For training manner, unsupervised methods generalize better than supervised methods, and we design an unsupervised early-stop strategy to help retain the best model with pre-trained weights as the basis. Extensive experiments are conducted to support the previous findings, on the basis of which we present an unsupervised stereo matching network with good generalization performance. We release the source code and the datasets at https://github.com/Elenairene/RKF_RSSM to reproduce the results and encourage future work.

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

↑