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

最新推荐文章于 2024-08-08 16:08:51 发布

just_sort

最新推荐文章于 2024-08-08 16:08:51 发布

阅读量1.1k

点赞数 2

分类专栏：数字图像处理论文和算法复现

本文链接：https://blog.csdn.net/just_sort/article/details/88710341

版权

数字图像处理论文和算法复现专栏收录该内容

53 篇文章 87 订阅

订阅专栏

算法原理

在这里插入图片描述

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);