Marr-Hildreth 边缘检测 OpenCV C++实现

算法原理

C++代码实现

Mat MarrEdgeDetection(Mat src, int kernelDiameter, double sigma) {
	int kernel_size = kernelDiameter / 2;
	Mat kernel(kernelDiameter, kernelDiameter, CV_64FC1);
	for (int i = -kernel_size; i <= kernel_size; i++) {
		for (int j = -kernel_size; j <= kernel_size; j++) {
			kernel.at<double>(i + kernel_size, j + kernel_size) = exp(-((pow(j, 2) + pow(i, 2)) /
				(pow(sigma, 2) * 2)))
				* (((pow(j, 2) + pow(i, 2) - 2 *
					pow(sigma, 2)) / (2 * pow(sigma, 4))));
		}
	}
	Mat laplacian(src.rows - kernel_size * 2, src.cols - kernel_size * 2, CV_64FC1);
	Mat dst = Mat::zeros(src.rows - kernel_size * 2, src.cols - kernel_size * 2, CV_8UC1);
	for (int i = kernel_size; i < src.rows - kernel_size; i++) {
		for (int j = kernel_size; j < src.cols - kernel_size; j++) {
			double sum = 0;
			for (int x = -kernel_size; x <= kernel_size; x++){
				for (int y = -kernel_size; y <= kernel_size; y++) {
					sum +=  src.at<uchar>(i + x, j + y) * kernel.at<double>(x + kernel_size, y + kernel_size);
				}
			}
			laplacian.at<double>(i - kernel_size, j - kernel_size) = sum;
		}
	}
	for (int i = 1; i < dst.rows - 1; i++) {
		for (int j = 1; j < dst.cols - 1; j++) {
			if ((laplacian.at<double>(i - 1, j) * laplacian.at<double>(i + 1, j) < 0) || (laplacian.at<double>(i, j + 1) * laplacian.at<double>(i, j - 1) < 0) ||
				(laplacian.at<double>(i + 1, j - 1) * laplacian.at<double>(i - 1, j + 1) < 0) || (laplacian.at<double>(i - 1, j - 1) * laplacian.at <double> (i + 1, j + 1) < 0)) {
				dst.at<uchar>(i, j) = 255;
			}
		}
	}
	return dst;
}
//调用
	Mat src = imread("F:\\1.jpg", 0);
	Mat dst = MarrEdgeDetection(src, 9, 1.6);

效果

原图：

结果图：