离散傅里叶变换

#include "stdafx.h"

#include <opencv2\highgui\highgui.hpp>
#include <opencv2\core\core.hpp>
#include <opencv2\imgproc\imgproc.hpp>
#include <iostream>
using namespace std;
using namespace cv;


int main()
{
    //【1】 以灰度模式读取原始图像并显示
    Mat srcImage = imread("D:\\pic\\demo2.jpg",0);
    if (!srcImage.data)
    {
        printf("error");
        system("pause");
        return -1;
    }
    imshow("原始图像", srcImage);


    cout << "原图像的行数：" << srcImage.rows << endl;
    cout << "原图像的列数：" << srcImage.cols << endl;
    cout << "原图像的通道数：" << srcImage.channels() << endl;

    int m = getOptimalDFTSize(srcImage.rows);
    int n = getOptimalDFTSize(srcImage.cols);

    cout << m << "  " << n << endl;

    //【2】将添加的像素初始化为0
    Mat padded;
    copyMakeBorder(srcImage, padded, 0, m - srcImage.rows, 0, n - srcImage.cols, BORDER_CONSTANT, Scalar::all(0));

    imshow("填充之后的图像", padded);

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


    Mat complexI;
    merge(planes,2, complexI);

    cout << complexI.rows<<" "<<complexI.cols<< " "<<complexI.channels()<<endl;
    //【4】 进行离散傅里叶变换
    dft(complexI, complexI);

    //【5】将复数转换为幅值，
    split(complexI, planes);//将多通道数组complexI分离成几个单通道数组，
    magnitude(planes[0], planes[1], planes[0]);
    Mat magnitudeImage = planes[0];

    //【6】进行对数尺度缩放
    magnitudeImage += Scalar::all(1); // M' = M +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));//ROI区域左上
    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);
    //【8】归一化
    normalize(magnitudeImage, magnitudeImage, 0, 1, CV_MINMAX);


    imshow("频谱幅值", magnitudeImage);

    waitKey();

    system("pause");
    return 0;
}

参考文献：
【1】、OpenCV3编程入门——毛星云