离散傅里叶变换

//---------离散傅里叶变换-----------------//
#include <iostream>
#include <opencv2/opencv.hpp>
#include <Windows.h>
#include <opencv2/highgui/highgui.hpp>
using namespace cv;
using namespace std;

int main(int argc,char *argv[],char **env)
{

	//一、以灰度图读入
	Mat srcImage = imread("C:/Users/hasee-pc/Desktop/dog.jpg",0);
	if (!srcImage.data)
	{
		printf("error in srcImage!\n");
		return -1;
	}
	//srcImage = Mat::zeros(500, 500,CV_8UC3);
	imshow("原始图像", srcImage);

	//二、将输入图像延展到最佳尺寸，边界用0填充9;
	int m = getOptimalDFTSize(srcImage.rows);
	int n = getOptimalDFTSize(srcImage.cols);
	//将添加的像素初始化0
	Mat padded;
	copyMakeBorder(srcImage, padded, 0, m - srcImage.rows, 0, n - srcImage.cols, BORDER_CONSTANT, Scalar::all(0));


	//三、为傅里叶变换的结果(实部和虚部）分配存储空间
	//将planes数组组合合并成一个多通道的数组complexI
	Mat planes[] = { Mat_<float>(padded),Mat::zeros(padded.size(),CV_32F) };
	Mat complexI;
	merge(planes, 2, complexI);

	//四、进行就地离散傅里叶变换
	dft(complexI, complexI);

	//五、将复数转换为幅值，即=>log(1+sqrt(Re(DFT(I))^2+Im(DFT(I))^2)
	split(complexI, planes);//将多通道数组complexI分离成几个单通道数组,
	//                        planes[0] = Re(DFT(I), planes[1] = Im(DFT(I));
	magnitude(planes[0], planes[1], planes[0]);//planes[0]=magnitude
	Mat magnitudeImage = planes[0];

	//六、进行对数尺度(logarithmic scale)缩放
	magnitudeImage += Scalar::all(1);
	log(magnitudeImage, magnitudeImage);//求自然对数

	//七、剪切和重分布幅度图象限
	//若有奇数行奇数列，进行频谱裁剪
	magnitudeImage = magnitudeImage(Rect(0, 0, magnitudeImage.cols&-2, magnitudeImage.rows &-2));

	//重新排列傅里叶图像中的象限，是的原点位于图像中心

	int cx = magnitudeImage.cols / 2;
	int cy = magnitudeImage.rows / 2;
	Mat q0(magnitudeImage, Rect(0, 0, cx, cy));
	Mat q1(magnitudeImage, Rect(cx, 0, cx, cy));
	Mat q2(magnitudeImage, Rect(0, cy, cx, cy));
	Mat q3(magnitudeImage, Rect(cx, cy, cx, cy));
	//交换象限（左上与右下进行交换）
	Mat tmp;
	q0.copyTo(tmp);
	q3.copyTo(q0);
	tmp.copyTo(q3);
	//交换象限(右上与左下进行交换）
	q1.copyTo(tmp);
	q2.copyTo(q1);
	tmp.copyTo(q2);

	//八、归一化，用0到1之间的浮点值将矩阵变换为可视的图像格式
	normalize(magnitudeImage, magnitudeImage, 0, 1, NORM_MINMAX);

	imshow("频谱幅值", magnitudeImage);
	waitKey(0);
	return 0;
}