c++ - CUDA 相当于 OpenCV 3 中的 estimateRigidTransform

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 等价物？

Answer 1

OpenCV 没有在 CUDA 上实现 estimateRigidTransform。 github 上有一个基于 opencv 的项目，它具有计算单应性和估计刚性变换的功能:https://github.com/danielsuo/cuSIFT

这是您需要的功能: https://github.com/danielsuo/cuSIFT/blob/master/extras/rigidTransform.cu