OpenCV 3 CUDA小测试

OpenCV 3版本的CUDA有了很大变化。刚刚在WIN7 64位系统编译好了带CUDA 7.0版本的OpenCV 3.0库。

进行了简单的测试。

CUDA版代码如下：

#include
#include
#include 
#include


using namespace std;
using namespace cv;


int main()
{
	clock_t t1 = clock();

	int num_devices = cv::cuda::getCudaEnabledDeviceCount();

	if (num_devices <= 0){
		cerr << "There is no device." << endl;
		return -1;
	}

	int enable_device_id = -1;
	for (int i = 0; i < num_devices; i++){
		cv::cuda::DeviceInfo dev_info(i);
		if (dev_info.isCompatible()){
			enable_device_id = i;
		}
	}

	if (enable_device_id < 0){
		cerr << "GPU module isn't built for GPU" << endl;
	}

	cv::cuda::setDevice(enable_device_id);

	Mat src_image = imread("Hi-RGB0971.tif");
	Mat dst_image;
	cuda::GpuMat d_src_img(src_image);//upload src image to gpu
	cuda::GpuMat d_dst_img;
	cuda::cvtColor(d_src_img, d_dst_img, CV_BGR2GRAY);
	d_dst_img.download(dst_image);
	imshow("test", dst_image);

	clock_t t2 = clock();

	std::cout << "time: " << (t2 - t1) / 1000 << "s" << endl;

	waitKey();

	return 0;

}

不用CUDA版：

#include
#include 
#include


using namespace std;
using namespace cv;


int main()
{
	clock_t t1 = clock();


	Mat src_image = imread("Hi-RGB0971.tif");
	Mat dst_image;

	cvtColor(src_image, dst_image, CV_BGR2GRAY);
	imshow("test", dst_image);

	clock_t t2 = clock();

	std::cout << "time: " << (t2 - t1) / 1000 << "s" << endl;

	waitKey();

	return 0;

}

测试下来，使用了CUDA编译的OpenCV库，运行的时间比没有使用CUDA库编译的OpenCV库还要长。我觉得是由于数据载入GPU和重新载入CPU耗费了时间。

后面会利用OpenCV库和CUDA单独写函数，分块处理影像，然后进行相关的测试。