边缘检测——Sobel

#include <opencv2/core/core.hpp>
#include <opencv2/highgui/highgui.hpp>
#include <iostream>
#include <cmath>

using namespace cv;
using namespace std;

int houghLineTransform(const char* str);
// Computes the x component of the gradient vector
// at a given point in a image.
// returns gradient in the x direction
int xGradient(Mat image, int x, int y)
{
	return image.at<uchar>(y-1, x-1) +
		2*image.at<uchar>(y, x-1) +
		image.at<uchar>(y+1, x-1) -
		image.at<uchar>(y-1, x+1) -
		2*image.at<uchar>(y, x+1) -
		image.at<uchar>(y+1, x+1);

	//  1  0  -1
	//  2  0  -2
	//  1  0  -1
}

// Computes the y component of the gradient vector
// at a given point in a image
// returns gradient in the y direction

int yGradient(Mat image, int x, int y)
{
	return image.at<uchar>(y-1, x-1) +
		2*image.at<uchar>(y-1, x) +
		image.at<uchar>(y-1, x+1) -
		image.at<uchar>(y+1, x-1) -
		2*image.at<uchar>(y+1, x) -
		image.at<uchar>(y+1, x+1);

	//  -1  -2  -1
	//  0    0   0
	//  1    2   1
}

int houghLineTransform(const char* str) {

	Mat src, dst;
	int gx, gy, sum;

	// Load an image
	src = imread(str, CV_LOAD_IMAGE_GRAYSCALE);
	dst = src.clone();
	if( !src.data )
	{ return -1; }


	for(int y = 0; y < src.rows; y++)
		for(int x = 0; x < src.cols; x++)
			dst.at<uchar>(y,x) = 0.0;

	for(int y = 1; y < src.rows - 1; y++){
		for(int x = 1; x < src.cols - 1; x++){
			gx = xGradient(src, x, y);
			gy = yGradient(src, x, y);
			sum = abs(gx) + abs(gy);
			sum = sum > 255 ? 255:sum;
			sum = sum < 0 ? 0 : sum;
			dst.at<uchar>(y,x) = sum;
		}
	}

	namedWindow("Sobel");
	imshow("Sobel", dst);

	namedWindow("initial");
	imshow("initial", src);

	return 0;
}

结果：