python - 如何在 C++ 中展平颜色直方图？

Question

您好，我用 python 编写了以下代码行:

# convert the image to HSV color-space
image = cv2.cvtColor(image, cv2.COLOR_BGR2HSV)

# compute the color histogram
hist  = cv2.calcHist([image], [0, 1, 2], None, [bins, bins, bins], [5, 240, 5, 240, 5, 240])

# normalize the histogram
cv2.normalize(hist, hist)

# return the histogram
return hist.flatten()

我现在正尝试用 C++ 重写它。我在 http://www.swarthmore.edu/NatSci/mzucker1/opencv-2.4.10-docs/doc/tutorials/imgproc/histograms/histogram_calculation/histogram_calculation.html 找到了一个很好的例子

我现在面临的问题是在 c++ 中展平 hist，例如在 python 代码中。这是展平 hist 在 python (512,) 中的输出形状。关于如何在 C++ 中获得相同结果的任何想法？

(编辑)到目前为止的 C++ 代码。

大小大小(500,500); image = imread("C:\johan.jpg",IMREAD_COLOR);

 resize(image,image,size);//resize image

 cvtColor(image, image, CV_BGR2HSV);
// Separate the image in 3 places ( H, S and V )
 vector<Mat> bgr_planes;
 split(image, bgr_planes );

 vector<Mat> hist_flat;

 // Establish the number of bins
 int histSize = 256;

 // Set the ranges ( for H,S,V) )
 float range[] = {5, 240} ;
 const float* histRange = { range };

 bool uniform = true; bool accumulate = false;

 Mat b_hist, g_hist, r_hist;

 cout << " Working fine Johan...";

 // Compute the histograms:
 calcHist( &bgr_planes[0], 1, 0, Mat(), b_hist, 1, &histSize, &histRange, uniform, accumulate );
 calcHist( &bgr_planes[1], 1, 0, Mat(), g_hist, 1, &histSize, &histRange, uniform, accumulate );
 calcHist( &bgr_planes[2], 1, 0, Mat(), r_hist, 1, &histSize, &histRange, uniform, accumulate );
 //calcHist( &image,3, 0, Mat(), hist_flat, 1, &histSize, &histRange, uniform, accumulate );



 // Draw the histograms for B, G and R
 int hist_w = 512; int hist_h = 400;
 int bin_w = cvRound( (double) hist_w/histSize );



 Mat histImage(hist_h,hist_w, CV_8UC3, Scalar(0,0,0));

 // Normalize the result to [ 0, histImage.rows ]
 normalize(b_hist, b_hist, 0, histImage.rows, NORM_MINMAX, -1, Mat() );
 normalize(g_hist, g_hist, 0, histImage.rows, NORM_MINMAX, -1, Mat() );
 normalize(r_hist, r_hist, 0, histImage.rows, NORM_MINMAX, -1, Mat() );



 // Draw for each channel
 for( int i = 1; i < histSize; i++ )
 {
     line( histImage, Point( bin_w*(i-1), hist_h - cvRound(b_hist.at<float>(i-1)) ) ,
                      Point( bin_w*(i), hist_h - cvRound(b_hist.at<float>(i)) ),
                      Scalar( 255, 0, 0), 2, 8, 0  );
     line( histImage, Point( bin_w*(i-1), hist_h - cvRound(g_hist.at<float>(i-1)) ) ,
                      Point( bin_w*(i), hist_h - cvRound(g_hist.at<float>(i)) ),
                      Scalar( 0, 255, 0), 2, 8, 0  );
     line( histImage, Point( bin_w*(i-1), hist_h - cvRound(r_hist.at<float>(i-1)) ) ,
                      Point( bin_w*(i), hist_h - cvRound(r_hist.at<float>(i)) ),
                      Scalar( 0, 0, 255), 2, 8, 0  );
 }

 // Display

 imshow("calcHist Demo", histImage );
 imshow("The image resized",image);

Answer 1

只是为这个问题添加一个答案。由于您使用 OpenCV cv::Mat 作为您的直方图持有者，一种使其变平的方法是使用 reshape 例如:

// create mat a with 512x512 size and float type
cv::Mat a(512,512,CV_32F);
// resize it to have only 1 row
a = a.reshape(0,1);

这个 O(1) 函数并且不复制元素，只是更改 cv::Mat header 以具有正确的大小。

在它之后你将有一个 1 行 cv::mat 和 262144 列。