边缘方向直方图

//边缘方向直方图（EOH）
//1、图像灰度化
//2、canny算法获得边缘图
//3、用sobel算子求灰度图像x、y方向梯度dx和dy
//4、计算各个像素的边缘方向theta(x,y) = atan(dy/dx)
//5、将边缘方向值进行量化，从-90-90量化到0-36
//6、将边缘方向theta进行直方图统计并归一化

#include "cv.h"
#include "highgui.h"
#include "iostream"

using namespace std;


int main()
{
IplImage *src = cvLoadImage("1.jpg", 1);
const int WIDTH = src->width;
const int HEIGHT = src->height;
CvMat *gray = cvCreateMat(HEIGHT, WIDTH, CV_8UC1);
cvCvtColor(src, gray, CV_BGR2GRAY);
//求解边缘图
CvMat *canny = cvCreateMat(HEIGHT, WIDTH, CV_8UC1);
cvCanny(gray, canny, 70, 170);
//分别求解x方向和y方向的梯度
CvMat *dx = cvCreateMat(HEIGHT, WIDTH, CV_8UC1);
CvMat *dy = cvCreateMat(HEIGHT, WIDTH, CV_8UC1);
cvSobel(gray, dx, 1, 0);
cvSobel(gray, dy, 0, 1);


CvMat *theta = cvCreateMat(HEIGHT, WIDTH, CV_32FC1);
int hist[36] = {0};
for(int j = 0; j < HEIGHT; j ++)
{
uchar *grayData = (uchar*)(gray->data.ptr + j * gray->step);
uchar *cannyData =  (uchar*)(canny->data.ptr + j * canny->step);
uchar *dxData =  (uchar*)(dx->data.ptr + j * dx->step);
uchar *dyData =  (uchar*)(dy->data.ptr + j * dy->step);
float *thetaData = (float*)(theta->data.ptr + j * theta->step);
for (int i = 0; i < WIDTH; i ++)
{
if (cannyData[i] == 0)
continue;
if (dxData[i] == 0)
{
if (dyData[i] > 0)
{
thetaData[i] = 90.0;
}
else if ( dyData[i] < 0)
{
thetaData[i] = - 90.0;
}
else
{
thetaData[i] = 0.0;
}

}
else
{
thetaData[i] = 180.0 / CV_PI * atan((float)dyData[i]/dxData[i]);
}


hist[(unsigned int)(((thetaData[i] + 90.0)- 0.5 ) / 5.0)] ++;
}
}


int maxValue = 0;
for (int i = 0; i < 36; i ++)
{
if (hist[i] > maxValue )
{
maxValue = hist[i];
}
}
IplImage *histImag = cvCreateImage(cvSize(360, 100), IPL_DEPTH_8U, 3);
cvZero(histImag);
for(int i = 0; i < 36; i++)
{
cvLine(histImag, cvPoint(i*10, 100), cvPoint(i*10, 100-(int)((float)hist[i] / maxValue * 100.0)), CV_RGB(255, 0, 0));
}
cvShowImage("SRC", src);
cvShowImage("HISTIMAGE", histImag);
cvWaitKey(0);
return 0;
}