手撕傅里叶

学习opencv之手撕傅里叶

借鉴一下

原理

二维离散傅立叶变换使用下式计算，其中I表示输入图像

二维离散傅立叶逆变换从频率分量 按照下式复原图像

代码复现

struct fourier_str {
	std::complex<double> coef[height][width];
};

// RGB to Gray scale
cv::Mat BGR2GRAY(cv::Mat img) {
	// prepare output
	cv::Mat out = cv::Mat::zeros(height, width, CV_8UC1);

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

// Discrete Fourier transformation
fourier_str dft(cv::Mat img, fourier_str fourier_s) {

	double I;
	double theta;
	std::complex<double> val;

	for (int l = 0; l < height; l++) {
		for (int k = 0; k < width; k++) {
			val.real(0);
			val.imag(0);
			for (int y = 0; y < height; y++) {
				for (int x = 0; x < width; x++) {
					I = (double)img.at<uchar>(y, x);
					theta = -2 * M_PI * ((double)k * (double)x / (double)width + (double)l * (double)y / (double)height);
					val += std::complex<double>(cos(theta), sin(theta)) * I;
				}
			}
			val /= sqrt(height * width);
			fourier_s.coef[l][k] = val;
		}
	}

	return fourier_s;
}

// Inverse Discrete Fourier transformation
cv::Mat idft(cv::Mat out, fourier_str fourier_s) {

	double theta;
	double g;
	std::complex<double> G;
	std::complex<double> val;

	for (int y = 0; y < height; y++) {
		for (int x = 0; x < width; x++) {
			val.real(0);
			val.imag(0);
			for (int l = 0; l < height; l++) {
				for (int k = 0; k < width; k++) {
					G = fourier_s.coef[l][k];
					theta = 2 * M_PI * ((double)k * (double)x / (double)width + (double)l * (double)y / (double)height);
					val += std::complex<double>(cos(theta), sin(theta)) * G;
				}
			}
			g = std::abs(val) / sqrt(height * width);
			g = fmin(fmax(g, 0), 255);
			out.at<uchar>(y, x) = (uchar)g;
		}
	}

	return out;
}

结果