用GPU调度优化无人机图像特征匹配,速度提升超77倍。
Efficient feature matching for UAV images based on compact GPU data scheduling
- 基于矩阵带宽压缩分块,动态生成紧凑图像块以提升GPU利用率。
- 相比KD树方法,特征匹配速度提升77至100倍,精度与传统方法相当。
- 适合处理大规模无人机影像数据,尤其适用于实时或资源受限场景。
特征匹配在结构光从运动(SfM)中占据主要计算耗时。本文提出一种面向无人机(UAV)图像的高效特征匹配GPU数据调度算法。核心思路是基于矩阵带宽压缩(MBR)将全数据集划分为图像块,通过GPU加速的级联哈希实现高效匹配。首先,利用图像检索技术将图像转换为全局描述符,结合图索引搜索高维最近邻;其次,基于MBR的数据调度策略迭代生成紧凑图像块,利用图像间连接关系提升GPU计算效率;第三,依据生成的图像块,在GPU加速级联哈希框架下顺序执行特征匹配,并通过局部几何约束与基于RANSAC的全局验证对初始匹配进行精修。为提升性能,匹配与验证步骤在GPU和CPU上并行执行。在大规模无人机数据集上的实验表明,该方法相较基于KD树的匹配方法,速度提升77.0至100.0倍,且在相对与绝对束调整(BA)中保持相当精度。所提算法为大规模无人机图像特征匹配提供高效解决方案。
原文摘要 · Abstract (English)
Feature matching dominates the time costs in structure from motion (SfM). The primary contribution of this study is a GPU data schedule algorithm for efficient feature matching of Unmanned aerial vehicle (UAV) images. The core idea is to divide the whole dataset into blocks based on matrix band reduction (MBR) and achieve efficient feature matching via GPU-accelerated cascade hashing. First, match pairs are selected by using an image retrieval technique, which converts images into global descriptors and searches high-dimension nearest neighbors with graph indexing. Second, compact image blocks are iteratively generated from a MBR-based data schedule strategy, which exploits image connections to generate image blocks and increase the usage of GPU computing power. Third, guided by the generated image blocks, feature matching is executed sequentially within the framework of GPU-accelerated cascade hashing, and initial candidate matches are refined by combining a local geometric constraint and RANSAC-based global verification. For further performance improvement, these two steps are designed to execute in parallel in GPU and CPU. Finally, the performance of the proposed solution is evaluated by using large-scale UAV datasets. The results demonstrate that it increases the efficiency of feature matching with speedup ratios ranging from 77.0 to 100.0 compared with KD-Tree based matching methods due to its high usage of GPU computing power. Besides, it achieves comparable accuracy in both relative and absolute bundle adjustment (BA). The proposed algorithm is an efficient solution for feature matching of large-scale UAV images.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。