颜色直方图特征提取

#include <cv.h>
#include <highgui.h>
#include <iostream>
#include <fstream>
using namespace std;

int main( int argc, char** argv )
{
	IplImage * src= cvLoadImage("testImgSet\\caimei.jpg");

	IplImage* hsv = cvCreateImage( cvGetSize(src), 8, 3 );
	/*
	HSV空间是直方图最常用的颜色空间。
	它的三个分量分别代表色彩（Hue）、饱和度（Saturation）和值（Value）
	*/
	//cvCreateImage(,,)
	IplImage* h_plane = cvCreateImage( cvGetSize(src), 8, 1 );
	IplImage* s_plane = cvCreateImage( cvGetSize(src), 8, 1 );
	IplImage* v_plane = cvCreateImage( cvGetSize(src), 8, 1 );
	IplImage* planes[] = { h_plane, s_plane };

	/** H 分量划分为16个等级，S分量划分为8个等级 */
	int h_bins = 16, s_bins = 8;
	int hist_size[] = {h_bins, s_bins};

	/** H 分量的变化范围 */
	float h_ranges[] = { 0, 180 }; 

	/** S 分量的变化范围*/
	float s_ranges[] = { 0, 255 };
	float* ranges[] = { h_ranges, s_ranges };

	/** 输入图像转换到HSV颜色空间 */
	cvCvtColor( src, hsv, CV_BGR2HSV );
	cvCvtPixToPlane( hsv, h_plane, s_plane, v_plane, 0 );
	//cvCvtPixToPlane()
	/** 创建直方图，二维, 每个维度上均分 */
	CvHistogram * hist = cvCreateHist( 2, hist_size, CV_HIST_ARRAY, ranges, 1 );
	/**
	/*typedef struct CvHistogram
               {
                  int type;
                  CvArr* bins;
                  float thresh[CV_MAX_DIM][2]; /* for uniform histograms 即均匀直方图得表示
                  float** thresh2; /* for non-uniform histograms 即非均匀直方图得表示
                  CvMatND mat; /* embedded matrix header for array histograms 
                }CvHistogram;//直方图的结构
         CvHistogram* cvCreateHist(
                 int     dims,//直方图维数的数目,如本例中表示2维直方图（分别是H和S）
                 int*    sizes,//整型数组得指针，数组的下标长度对应直方图得维数（dims），如本例中表示H维的刻度分为16个，S为8个
                 int     type,//对应上面的type，如本例中表示CV_HIST_ARRAY
                 float** ranges  = NULL,//直方图各维对应的范围，如本例中表示H维中表示范围是{0,255}
                 接上一行(0 到255对应直方图结构中的float thresh[CV_MAX_DIM][2]; 中得2（CV_MAX_DIM表示直方图能表示得维数得最大值）
                 int     uniform = 1//均匀直方图的表示
                                  );这个函数个人认为是个难点。
	*/
	/** 根据H,S两个平面数据统计直方图 */
	cvCalcHist( planes, hist, 0, 0 );
	
	/** 获取直方图统计的最大值，用于动态显示直方图 */
	float max_value;
	cvGetMinMaxHistValue( hist, 0, &max_value, 0, 0 );
	//统计直方图中最大值，赋给max_value所指单元
	//（对应的最大值对应的横坐标单元的高度为240（即纵坐标得长度））

	/** 设置直方图显示图像 */
	int height = 240;//纵坐标长240个像素长度（到后面可知每个刻度长为6个像素长度）
	int width = (h_bins*s_bins*6); //横坐标长16*8*6个像素长度（到后面可知每个刻度长为6个像素长度）
	IplImage* hist_img = cvCreateImage( cvSize(width,height), 8, 3 );
	cvZero( hist_img );

	/** 用来进行HSV到RGB颜色转换的临时单位图像 */
	IplImage * hsv_color = cvCreateImage(cvSize(1,1),8,3);
	IplImage * rgb_color = cvCreateImage(cvSize(1,1),8,3);
	int bin_w = width / (h_bins * s_bins);//个刻度长为6个像素长度
	
	//说明 本来 二维图是 横坐标为S  纵坐标为H
	//但本例 但此例输出得图表示为：横坐标表示HSV颜色空间中HS各占的刻度，
	//即（H刻度，S刻度）
	//H刻度从0增到16，S刻度从0增到8;
	//纵坐标表示对应各个HS所占比例
	//freopen("ColorHis.txt","w",stdout);
	for(int h = 0; h < h_bins; h++)
	{
		for(int s = 0; s < s_bins; s++)
		{
			int i = h*s_bins + s;
			/** 获得直方图中的统计次数，计算显示在图像中的高度 */
			float bin_val = cvQueryHistValue_2D( hist, h, s );
			int intensity = cvRound(bin_val*height/max_value);//强度
			//cvRound 对一个double型的数进行四舍五入，并返回一个整型数
			//cout<<bin_val<<" "<<intensity<<" ";
			/** 获得当前直方图代表的颜色，转换成RGB用于绘制 */
			cvSet2D(hsv_color,0,0,cvScalar(h*180.f / h_bins,s*255.f/s_bins,255,0));
			cvCvtColor(hsv_color,rgb_color,CV_HSV2BGR);
			CvScalar color = cvGet2D(rgb_color,0,0);

			cvRectangle( hist_img, cvPoint(i*bin_w,height),
				cvPoint((i+1)*bin_w,height - intensity),
				color, -1, 8, 0 );
		}
		//cout<<endl;
	}
	//fclose(stdout);
	cvNamedWindow( "Source", 1 );
	cvShowImage( "Source", src );

	cvNamedWindow( "H-S Histogram", 1 );
	cvShowImage( "H-S Histogram", hist_img );

	cvWaitKey(0);
}