OpenCV 3中estimateRigidTransform的CUDA等效版本

Question

我正在使用OpenCV进行视频稳定项目的开发，已经实现了CPU版本，但是性能需要改进，因此我正在尝试将大部分处理移至GPU上。

目前的实现主要使用以下四个OpenCV函数：

- cv::goodFeaturesToTrack - cv::calcOpticalFlowPyrLK - cv::estimateRigidTransform - cv::warpAffine 到目前为止，我已在GPU上找到了以下等效函数：

- cv::cuda::createGoodFeaturesToTrackDetector - cv::cuda::SparsePyrLKOpticalFlow - cv::cuda::warpAffine 是否有estimateRigidTransform的CUDA等价函数？

- Alex L

---

我不确定你是否需要这个。在 CPU 上速度应该很快，将其加载到 GPU 的开销可能会使低点数的速度变慢。estimateRigidTransform 使用最小二乘法，可以使用 cuda cuSolver 计算。如果你真的想在 GPU 上实现它，那么实现起来应该不难。 - taarraas

嘿@taarraas，感谢您的评论。我正在重写视频稳定功能，以便100％在GPU上运行，因此我不想下载/上传点以在CPU上运行estimateRigidTransform。如果GPU上没有现有的实现，那么我将使用CUDA编写它，只是想问一下，以防我错过了什么 :) - Alex L

Answer 1

OpenCV没有CUDA的estimateRigidTransform实现。在GitHub上有一个基于OpenCV的项目，其中包含计算单应性和估计刚性变换的函数：https://github.com/danielsuo/cuSIFT。

以下是您需要的函数： https://github.com/danielsuo/cuSIFT/blob/master/extras/rigidTransform.cu。

- taarraas