如何从 opencv gpumat 创建推力 device_vector

Question

我需要将 opencv::gpumat 转换为推力::device_vector，然后执行推力::copy_if，然后将生成的推力::device_vector 再次复制到 cv::gpumat。

如何从 cv::gpumat 转换为推力::device_vector 反之亦然？

我的代码（如下）给出了编译错误： 错误：需要一个类型说明符

gcc (Ubuntu 5.4.0-6ubuntu1~16.04.10) 5.4.0 20160609

nvcc：Cuda 编译工具，8.0 版，V8.0.44

推力 v1.8

ubuntu 16.04

我的代码：

#include <thrust/random.h>

#include <thrust/device_ptr.h>
#include <thrust/device_vector.h>

#include <opencv2/core/cuda.hpp>

int main(void)
{
    cv::Mat input_host(1,100,CV_32F);
    cv::Mat output_host(1,100,CV_32F);

    cv::randu(input_host, cv::Scalar(-10), cv::Scalar(10));
    cv::cuda::GpuMat input_device(1, 100, CV_32F);
    input_device.upload(input_host);

    thrust::device_ptr<CV_32F> input_ptr(input_device.ptr());

    thrust::device_vector<CV_32F>   input_device_vector  (input_ptr,100);
    return 0;
}

Answer 1

如何从 cv::gpumat 转换为推力::device_vector ....

这是无法做到的。 Thrust 没有公开从现有指针构造device_vector 的机制。复制构造是 device_vector 可以从现有数据实例化的唯一方法。

.... 反之亦然？

这是可以做到的：

1. 将device_vector 强制转换为原始指针（请参阅here）
2. 从原始指针创建一个 openCV gpumat 实例（参见here）

请注意，如果您让device_vector 超出范围，则会调用其析构函数，并且支持它的内存（以及使用相同内存的任何其他内容）将被释放，并且您处于未定义的行为领域。