2016-11-22 51 views
1

我试图做一个基于中心的直方图,这意味着在中间的RGB图像的值更重要。我写了下面的代码,但根据OpenCV自己的直方图函数,它非常慢。有什么办法可以让这个速度变快吗?基于OpenCV中心的直方图

void calc_histogram(cv::Mat& image, cv::Mat& histogram) 
{ 
int bit = 8; 
int max_value = pow(2, bit); 

int n_blue_bin = 8; 
int n_green_bin = 8; 
int n_red_bin = 8; 

int blue_width = max_value/n_blue_bin; 
int green_width = max_value/n_green_bin; 
int red_width = max_value/n_red_bin; 

int mySizes[3]={n_blue_bin, n_green_bin, n_red_bin}; 
histogram = Mat::zeros(3,mySizes,CV_32F); 

int blue, green, red; 
int blue_bin, green_bin, red_bin; 

float k, length; 
float max_length = sqrt((image.rows/2)*(image.rows/2) + (image.cols/2)*(image.cols/2)); 

for(int i = 0 ; i < image.rows ; i++) 
{ 
    for(int j = 0 ; j < image.cols ; j++) 
    { 
     length = sqrt(abs(image.rows/2 - i)*abs(image.rows/2 - i) + abs(image.cols/2 - j)*abs(image.cols/2 - j))/max_length; 
     k = sqrt(1-length); 

     Vec3b intensity = image.at<Vec3b>(i, j); 
     blue = intensity.val[0]; 
     green = intensity.val[1]; 
     red = intensity.val[2]; 

     blue_bin = blue/blue_width; 
     green_bin = green/green_width; 
     red_bin = red/red_width; 
     histogram.at<float>(blue_bin, green_bin, red_bin) += k; 
    } 
} 
} 
+0

你是否分析了你的代码?这通常是一个好的开始。我怀疑在你的内部循环中对'sqrt()'的两次调用可能会成为你的主要嫌疑人... –

+0

@TobySpeight他们可能是,但我认为他们是必要的。 –

回答

0

这对我来说不是一个直方图函数:我的结果是否应该是一个8x8x8矩阵?通常直方图是3xNumBins。要添加,请执行直方图[0] [red] + = k;直方图[1] [绿色] + = K;直方图[2] [蓝色] + = K;通常k = 1,但在你的情况下,它是一个重量。也许我正在阅读opencv错误。

此外,您正在做很多多余的工作:为什么在平方之前的abs?当你可以乘以倒数时,不要分裂。迭代y然后x。不要使用at,而是使用指针。

+0

是的,我得到一个8x8x8矩阵,没有关于这个值的问题。是的,我正在做一些与通常直方图不同的东西。你对abs()是正确的。我会修好它。但我没有得到相互的和部分的。 –

+0

@JennySimon互惠:乘以1/x要快于除以x要快得多。预先计算这些。在:使用.ptr(行)而不是at。翻转你的循环去内循环中的列。然后每行执行一次ptr并索引myptr [col]。还要重新考虑你对sqrt的使用。你不能只用方形而不是两根? – starmole