大津阈值Opencv实现

问题一 灰度化

Mat BGR2GRAY(Mat img) {
	// get height and width
	int width = img.cols;
	int height = img.rows;

	// prepare output
	Mat out = Mat::zeros(height, width, CV_8UC1);

	// each y, x
	for (int y = 0; y < height; y++) {
		for (int x = 0; x < width; x++) {
			// BGR -> Gray
			out.at<uchar>(y, x) = 0.2126 * (float)img.at<Vec3b>(y, x)[2] \
				+ 0.7152 * (float)img.at<Vec3b>(y, x)[1] \
				+ 0.0722 * (float)img.at<Vec3b>(y, x)[0];
		}
	}

	return out;
}

使用大津算法实现分割

Mat Binarize_Otsu(Mat gray) {
	int width = gray.cols;
	int height = gray.rows;

	// determine threshold
	double w0 = 0, w1 = 0;
	double m0 = 0, m1 = 0;
	double max_sb = 0, sb = 0;
	int th = 0;
	int val;

	// Get threshold
	for (int t = 0; t < 156; t++) {
		w0 = 0;
		w1 = 0;
		m0 = 0;
		m1 = 0;
		for (int y = 0; y < height; y++) {
			for (int x = 0; x < width; x++) {
				val = (int)(gray.at<uchar>(y, x));

				if (val < t) {
					w0++;
					m0 += val;
				}
				else {
					w1++;
					m1 += val;
				}
			}
		}

		m0 /= w0;
		m1 /= w1;
		w0 /= (height * width);
		w1 /= (height * width);
		sb = w0 * w1 * pow((m0 - m1), 2);

		if (sb > max_sb) {
			max_sb = sb;
			th = t;
		}
	}

	std::cout << "threshold:" << th << std::endl;

	// prepare output
	Mat out = Mat::zeros(height, width, CV_8UC1);

	// each y, x
	for (int y = 0; y < height; y++) {
		for (int x = 0; x < width; x++) {
			// Binarize
			if (gray.at<uchar>(y, x) > th) {
				out.at<uchar>(y, x) = 255;
			}
			else {
				out.at<uchar>(y, x) = 0;
			}

		}
	}

	return out;
}

成果展示


平均池化：

Mat average_pooling(Mat img) {
	int height = img.rows;
	int width = img.cols;
	int channel = img.channels();

	// prepare output
	Mat out = Mat::zeros(height, width, CV_8UC3);

	int r = 8;
	double v = 0;

	for (int y = 0; y < height; y += r) {
		for (int x = 0; x < width; x += r) {
			for (int c = 0; c < channel; c++) {
				v = 0;
				for (int dy = 0; dy < r; dy++) {
					for (int dx = 0; dx < r; dx++) {
						v += (double)img.at<Vec3b>(y + dy, x + dx)[c];
					}
				}
				v /= (r * r);
				for (int dy = 0; dy < r; dy++) {
					for (int dx = 0; dx < r; dx++) {
						out.at<Vec3b>(y + dy, x + dx)[c] = (uchar)v;
					}
				}
			}
		}
	}
	return out;
}