gpu :: morphologyEx比CPU中的morphologyEx慢?

时间:2014-01-10 21:38:24

标签: c++ opencv cuda gpu mathematical-morphology

我正在编写一个c ++代码,用于比较使用CPU和GPU版本的openscv的morphologyEx方法的性能。这是我的代码:

#include <opencv2/opencv.hpp>
#include <opencv2/gpu/gpu.hpp>
#include <sys/time.h>       
#include <ctime>
using namespace cv;
using namespace std;


double start_timer()
{
     double start_time = (double) getTickCount();
     return start_time;
}

double end_timer(double start_time,int num_tests)
{
    double time = (1000 * ((double) getTickCount() - start_time)/ getTickFrequency());
    cout << "Average time of " << num_tests  << " frames is: " << time/num_tests <<  " ms" << endl;
    return time;
}


int main()
{
    Mat cpuSrc;
    cv::gpu::GpuMat src_gpu, dst_gpu;
    Mat dst;
    Mat element;
    int element_shape = MORPH_RECT;
    element = getStructuringElement(element_shape, Size(10, 10 ), Point(-1, -1) );
    cpuSrc = imread("images.jpeg",CV_LOAD_IMAGE_ANYDEPTH);

    if (!cpuSrc.data)
    {
        cerr << "Cannot read the data" << endl;
        return -1;
    }


    cout << "Starting calculating time for CPU ....." << endl;
    double start_time = start_timer();
    int d = 0;
    while(d<100)
    {
        cv::morphologyEx(cpuSrc, dst, CV_MOP_OPEN, element,Point(-1,-1),1);
    }

    double total_time_cpu = end_timer(start_time,d);



//--------------------------------------------------------------
    cout << "Starting calculating time for GPU ....." << endl;
    d = 0;
    cv::gpu::GpuMat buf1, buf2;
    gpu::Stream stream;
    double start_time_1 = start_timer();

    while(d<100)
    {
        stream.enqueueUpload(cpuSrc, src_gpu);
        cv::gpu::morphologyEx(src_gpu,dst_gpu,CV_MOP_OPEN,element,
                   buf1,buf2,Point(-1,-1),1,stream);
        stream.enqueueDownload(dst_gpu, dst);

    }
    stream.waitForCompletion();
    double total_time_gpu = end_timer(start_time_1,d);

    cout << "Gain is: " << total_time_cpu / total_time_gpu << endl;
    return 0;
}
  
    

我正在使用循环,好像我正在模拟包含100帧的视频。我使用的是NVIDIA Corporation GF110 [GeForce GTX 570]和Intel Corporation Xeon E5 / Core i7 DMI2。此外,我测试了上传和下载的时间,并且它在第一帧中非常大但是之后可以忽略大约上传它是每帧0.02ms并且下载是0.1ms并且主要时间消耗是使用morphologyEx操作。

  

此模拟的时间结果如下:

  

用于CPU形态版本,The    100帧的平均时间是:: 0.027349 ms,GPU版本的平均时间是:: 18.0128 ms

     
    

请你帮我弄清楚出现这种意外表现的原因是什么?!!

  

提前非常感谢你。

1 个答案:

答案 0 :(得分:1)

在初始化中,您应该调用:

cv::gpu::setDevice(0);

它将加速初始化。