天天看点

关于cuda和opencv混合编译之_双边滤波

说起双边滤波,主要这个算法在cpu上太耗时了,平均下来,要100多毫秒.

怎么办呢,不怕,加速..

bool bilateralFilter_gpu(const cv::Mat& src,cv::Mat &dst)
{
	if (src.data == nullptr)
	{
		return false;
	}

//这里不介绍了,日常的上传下载.

	cv::cuda::GpuMat src_gpu, dst_gpu;
	src_gpu.upload(src);

//注意这里,这个算子,这几个参数如果设置不对,这个算法是会奔溃的.好在我们一般使用下面那个现成的.
	//cv::Ptr<cv::cuda::DisparityBilateralFilter> bilateralKernel =         
    // cv::cuda::createDisparityBilateralFilter(63,27,55);
	//bilateralKernel->apply(src_gpu, dst_gpu);

	//注意这里两个参数的取值范围.
	float sigma_color = 10;
	float sigma_spatial = 10;
	cv::cuda::bilateralFilter(src_gpu,dst_gpu,3, sigma_color, sigma_spatial);

	if (dst_gpu.data != nullptr)
	{
		dst_gpu.download(dst);
	}
	else
	{
		qDebug() << QStringLiteral("计算失败");
	}
	
	return true;
}

void main()
{

    //日常测试加速.
cv::Mat bilateral_result = cv::Mat::zeros(filter_src.cols, filter_src.filter_src.type());
	bilateralFilter_gpu(filter_src, bilateral_result);

}
           

这个经过加速,其时间 缩减到了 3ms,不过这个看你加速的图像,如果分辨率比较高的,那时间相对长一点

继续阅读