opencv之求直线的角度

opencv之求直线的角度，这也是《learning opencv》中第六章的练习5

 

#include <iostream>
#include <cv.h>
#include <highgui.h>
#include <cxcore.h>

using namespace std;
const CvSize size = cvSize(200,200);
const int aperture[] = {3,5,9,11,13,17};
int main()
{
	IplImage *src = cvCreateImage(size,8,1);
	cvZero(src);
	cvLine(src
		//,cvPoint(0,size.height - 1)
		//,cvPoint(size.width -1 , 0)
		,cvPoint(0,0)
		,cvPoint(size.width - 1, size.height - 1)
		,CV_RGB(255,255,255)
		,3
		);
	cvShowImage("src",src);
	
	IplImage *deriv_x = cvCreateImage(size,IPL_DEPTH_32F,1);
	IplImage *deriv_y = cvCreateImage(size,IPL_DEPTH_32F,1);
	IplImage *magnitude = cvCreateImage(size,IPL_DEPTH_32F,1);
	IplImage *angle = cvCreateImage(size,IPL_DEPTH_32F,1);
	IplImage *mask = cvCreateImage(size,IPL_DEPTH_8U,1);
	for(int i=0;i<sizeof(aperture)/sizeof(aperture[0]);++i){
		cvSobel(src,deriv_x,1,0,aperture[i]);
		cvSobel(src,deriv_y,0,1,aperture[i]);

		cvCartToPolar(deriv_x,deriv_y,magnitude,angle,1);

		cvSave("x.xml",deriv_x);
		cvSave("y.xml",deriv_y);
		cvSave("magnitude.xml",magnitude);
		cvSave("angle.xml",angle);

		double max_mag ,min_mag ,max_angle , min_angle;
		cvMinMaxLoc(magnitude,&min_mag,&max_mag);
		cvMinMaxLoc(angle,&min_angle,&max_angle);
		cout<<"magnitude: max = "<<max_mag<<" min = "<<min_mag<<endl;
		cout<<"angle : max = "<<max_angle<<" min = "<<min_angle<<endl;
		
		cvCmpS(magnitude,max_mag*3/4,mask,CV_CMP_GT);
		cvShowImage("mask",mask);

		CvScalar scalar = cvAvg(angle,mask);
		cout<<"aperture = "<<aperture[i]<<", line angle = "<<scalar.val[0]<<endl;
		cvWaitKey();
	}
	cvReleaseImage(&src);
	cvReleaseImage(&magnitude);
	cvReleaseImage(&angle);
	cvReleaseImage(&mask);
	cvDestroyAllWindows();

}