Opencv C++ 离散傅里叶变换
输入图片下载地址:
链接: https://pan.baidu.com/s/1KAVNvCRuR8KY2dseU5eaBQ 提取码: 1axt
#include <opencv2/core.hpp>
#include <opencv2/highgui.hpp>
#include <opencv2/imgproc.hpp>
#include <iostream>
using namespace std;
using namespace cv;
int main()
{
//1.读取并显示图片
Mat img_src = imread("../Data/5_5_image.jpg", IMREAD_GRAYSCALE);
imshow("img_src", img_src);
//2.输入图片扩充尺寸
int m = getOptimalDFTSize(img_src.rows);
int n = getOptimalDFTSize(img_src.cols);
Mat img_padded;
copyMakeBorder(img_src, img_padded, 0, m - img_src.rows, 0, n - img_src.cols, BORDER_CONSTANT, Scalar::all(0));
cout << "扩充图片 高度: " << img_padded.rows << " 宽度:" << img_padded.cols;
//3.为傅里叶变换结果分配存储空间
Mat planes[] = { Mat_<float>(img_padded), Mat::zeros(img_padded.size(), CV_32F) };
Mat complexI;
merge(planes, 2, complexI);
//4.执行傅里叶变换
dft(complexI, complexI);
//5.将复数转换为幅值
split(complexI, planes);
magnitude(planes[0], planes[1], planes[0]);
Mat magnitudeImage = planes[0];
//6.进行对数尺度缩放
magnitudeImage += Scalar::all(1);
log(magnitudeImage, magnitudeImage); //自然对数
//7.重新分布幅度图像象限
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 temp;
q0.copyTo(temp);
q3.copyTo(q0);
temp.copyTo(q3);
q1.copyTo(temp);
q2.copyTo(q1);
temp.copyTo(q2);
//8.归一化显示
normalize(magnitudeImage, magnitudeImage, 0, 1, NORM_MINMAX);
//9.显示频谱图片
imshow("频谱幅值", magnitudeImage);
waitKey();
destroyAllWindows();
return 0;
}