parent
08724393f4
commit
6d549490fb
1 changed files with 81 additions and 0 deletions
@ -0,0 +1,81 @@ |
||||
/**
|
||||
* @function calcHist_Demo.cpp |
||||
* @brief Demo code to use the function calcHist |
||||
* @author |
||||
*/ |
||||
|
||||
#include "opencv2/highgui/highgui.hpp" |
||||
#include "opencv2/imgproc/imgproc.hpp" |
||||
#include <iostream> |
||||
#include <stdio.h> |
||||
|
||||
using namespace std; |
||||
using namespace cv; |
||||
|
||||
/**
|
||||
* @function main |
||||
*/ |
||||
int main( int argc, char** argv ) |
||||
{ |
||||
Mat src, dst; |
||||
|
||||
/// Load image
|
||||
src = imread( argv[1], 1 ); |
||||
|
||||
if( !src.data ) |
||||
{ return -1; } |
||||
|
||||
/// Separate the image in 3 places ( R, G and B )
|
||||
vector<Mat> rgb_planes; |
||||
split( src, rgb_planes );
|
||||
|
||||
/// Establish the number of bins
|
||||
int histSize = 255; |
||||
|
||||
/// Set the ranges ( for R,G,B) )
|
||||
float range[] = { 0, 255 } ;
|
||||
const float* histRange = { range }; |
||||
|
||||
bool uniform = true; bool accumulate = false; |
||||
|
||||
Mat r_hist, g_hist, b_hist; |
||||
|
||||
/// Compute the histograms:
|
||||
calcHist( &rgb_planes[0], 1, 0, Mat(), r_hist, 1, &histSize, &histRange, uniform, accumulate ); |
||||
calcHist( &rgb_planes[1], 1, 0, Mat(), g_hist, 1, &histSize, &histRange, uniform, accumulate ); |
||||
calcHist( &rgb_planes[2], 1, 0, Mat(), b_hist, 1, &histSize, &histRange, uniform, accumulate ); |
||||
|
||||
// Draw the histograms for R, G and B
|
||||
int hist_w = 400; int hist_h = 400; |
||||
int bin_w = cvRound( (double) hist_w/histSize ); |
||||
|
||||
Mat histImage( hist_w, hist_h, CV_8UC3, Scalar( 0,0,0) ); |
||||
|
||||
/// Normalize the result to [ 0, histImage.rows ]
|
||||
normalize(r_hist, r_hist, 0, histImage.rows, NORM_MINMAX, -1, Mat() ); |
||||
normalize(g_hist, g_hist, 0, histImage.rows, NORM_MINMAX, -1, Mat() ); |
||||
normalize(b_hist, b_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(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 );
|
||||
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(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 );
|
||||
} |
||||
|
||||
/// Display
|
||||
namedWindow("calcHist Demo", CV_WINDOW_AUTOSIZE ); |
||||
imshow("calcHist Demo", histImage ); |
||||
|
||||
waitKey(0); |
||||
|
||||
return 0; |
||||
|
||||
} |
Loading…
Reference in new issue