opencv傅里叶变换实例

来源:互联网 发布:博客源码手机 编辑:程序博客网 时间:2024/06/06 01:58

OpenCV快速傅里叶变换实例

int main(){    Mat I = imread("ted_cruz.jpg", CV_LOAD_IMAGE_GRAYSCALE);    if (I.empty())        return -1;    cout << I.size() << endl;    Mat padded;                            //expand input image to optimal size    int m = getOptimalDFTSize(I.rows);    int n = getOptimalDFTSize(I.cols); // on the border add zero values    copyMakeBorder(I, padded, 0, m - I.rows, 0, n - I.cols, BORDER_CONSTANT, Scalar::all(0));    cout << padded.size() << endl;    Mat planes[] = { Mat_<float>(padded), Mat::zeros(padded.size(), CV_32F) };    Mat complexI;    merge(planes, 2, complexI);         // Add to the expanded another plane with zeros    dft(complexI, complexI);            // this way the result may fit in the source matrix    // compute the magnitude and switch to logarithmic scale    // => log(1 + sqrt(Re(DFT(I))^2 + Im(DFT(I))^2))    split(complexI, planes);                   // planes[0] = Re(DFT(I), planes[1] = Im(DFT(I))    magnitude(planes[0], planes[1], planes[0]);// planes[0] = magnitude    Mat magI = planes[0];    magI += Scalar::all(1);                    // switch to logarithmic scale    log(magI, magI);    // crop the spectrum, if it has an odd number of rows or columns    magI = magI(Rect(0, 0, magI.cols & -2, magI.rows & -2));    // rearrange the quadrants of Fourier image  so that the origin is at the image center    int cx = magI.cols / 2;    int cy = magI.rows / 2;    Mat q0(magI, Rect(0, 0, cx, cy));   // Top-Left - Create a ROI per quadrant    Mat q1(magI, Rect(cx, 0, cx, cy));  // Top-Right    Mat q2(magI, Rect(0, cy, cx, cy));  // Bottom-Left    Mat q3(magI, Rect(cx, cy, cx, cy)); // Bottom-Right    Mat tmp;                           // swap quadrants (Top-Left with Bottom-Right)    q0.copyTo(tmp);    q3.copyTo(q0);    tmp.copyTo(q3);    q1.copyTo(tmp);                    // swap quadrant (Top-Right with Bottom-Left)    q2.copyTo(q1);    tmp.copyTo(q2);    normalize(magI, magI, 0, 1, CV_MINMAX); // Transform the matrix with float values into a    // viewable image form (float between values 0 and 1).    imshow("Input Image", I);    // Show the result    imshow("spectrum magnitude", magI);    waitKey();    return 0;}OpenCV 中 傅里叶变换 FFT,代码如下:

[cpp] view plain copy
  1. void fft2(IplImage *src, IplImage *dst)  
  2. {   //实部、虚部  
  3.     IplImage *image_Re = 0, *image_Im = 0, *Fourier = 0;  
  4.     //   int i, j;  
  5.     image_Re = cvCreateImage(cvGetSize(src), IPL_DEPTH_64F, 1);  //实部  
  6.     //Imaginary part  
  7.     image_Im = cvCreateImage(cvGetSize(src), IPL_DEPTH_64F, 1);  //虚部  
  8.     //2 channels (image_Re, image_Im)  
  9.     Fourier = cvCreateImage(cvGetSize(src), IPL_DEPTH_64F, 2);  
  10.     // Real part conversion from u8 to 64f (double)  
  11.     cvConvertScale(src, image_Re);  
  12.     // Imaginary part (zeros)  
  13.     cvZero(image_Im);  
  14.     // Join real and imaginary parts and stock them in Fourier image  
  15.     cvMerge(image_Re, image_Im, 0, 0, Fourier);  
  16.   
  17.     // Application of the forward Fourier transform  
  18.     cvDFT(Fourier, dst, CV_DXT_FORWARD);  
  19.     cvReleaseImage(&image_Re);  
  20.     cvReleaseImage(&image_Im);  
  21.     cvReleaseImage(&Fourier);  
  22. }  
  23.   
  24. void fft2shift(IplImage *src, IplImage *dst)  
  25. {  
  26.     IplImage *image_Re = 0, *image_Im = 0;  
  27.     int nRow, nCol, i, j, cy, cx;  
  28.     double scale, shift, tmp13, tmp24;  
  29.     image_Re = cvCreateImage(cvGetSize(src), IPL_DEPTH_64F, 1);  
  30.     //Imaginary part  
  31.     image_Im = cvCreateImage(cvGetSize(src), IPL_DEPTH_64F, 1);  
  32.     cvSplit( src, image_Re, image_Im, 0, 0 );  
  33.     //具体原理见冈萨雷斯数字图像处理p123  
  34.     // Compute the magnitude of the spectrum Mag = sqrt(Re^2 + Im^2)  
  35.     //计算傅里叶谱  
  36.     cvPow( image_Re, image_Re, 2.0);  
  37.     cvPow( image_Im, image_Im, 2.0);  
  38.     cvAdd( image_Re, image_Im, image_Re);  
  39.     cvPow( image_Re, image_Re, 0.5 );  
  40.     //对数变换以增强灰度级细节(这种变换使以窄带低灰度输入图像值映射  
  41.     //一宽带输出值,具体可见冈萨雷斯数字图像处理p62)  
  42.     // Compute log(1 + Mag);  
  43.     cvAddS( image_Re, cvScalar(1.0), image_Re ); // 1 + Mag  
  44.     cvLog( image_Re, image_Re ); // log(1 + Mag)  
  45.   
  46.     //Rearrange the quadrants of Fourier image so that the origin is at the image center  
  47.     nRow = src->height;  
  48.     nCol = src->width;  
  49.     cy = nRow/2; // image center  
  50.     cx = nCol/2;  
  51.     //CV_IMAGE_ELEM为OpenCV定义的宏,用来读取图像的像素值,这一部分就是进行中心变换  
  52.     for( j = 0; j < cy; j++ ){  
  53.         for( i = 0; i < cx; i++ ){  
  54.             //中心化,将整体份成四块进行对角交换  
  55.             tmp13 = CV_IMAGE_ELEM( image_Re, double, j, i);  
  56.             CV_IMAGE_ELEM( image_Re, double, j, i) = CV_IMAGE_ELEM(  
  57.                 image_Re, double, j+cy, i+cx);  
  58.             CV_IMAGE_ELEM( image_Re, double, j+cy, i+cx) = tmp13;  
  59.   
  60.             tmp24 = CV_IMAGE_ELEM( image_Re, double, j, i+cx);  
  61.             CV_IMAGE_ELEM( image_Re, double, j, i+cx) =  
  62.                 CV_IMAGE_ELEM( image_Re, double, j+cy, i);  
  63.             CV_IMAGE_ELEM( image_Re, double, j+cy, i) = tmp24;  
  64.         }  
  65.     }  
  66.     //归一化处理将矩阵的元素值归一为[0,255]  
  67.     //[(f(x,y)-minVal)/(maxVal-minVal)]*255  
  68.     double minVal = 0, maxVal = 0;  
  69.     // Localize minimum and maximum values  
  70.     cvMinMaxLoc( image_Re, &minVal, &maxVal );  
  71.     // Normalize image (0 - 255) to be observed as an u8 image  
  72.     scale = 255/(maxVal - minVal);  
  73.     shift = -minVal * scale;  
  74.     cvConvertScale(image_Re, dst, scale, shift);  
  75.     cvReleaseImage(&image_Re);  
  76.     cvReleaseImage(&image_Im);  
  77. }  
  78.   
  79. void CCVMFCView::OnFuliyeTransform()  
  80. {  
  81.     IplImage *src;  
  82.     IplImage *Fourier;   //傅里叶系数  
  83.     IplImage *dst ;  
  84.     IplImage *ImageRe;  
  85.     IplImage *ImageIm;  
  86.     IplImage *Image;  
  87.     IplImage *ImageDst;  
  88.     double m,M;  
  89.     double scale;  
  90.     double shift;  
  91.     //src = workImg;  
  92.     if(workImg->nChannels==3)  
  93.         OnColorToGray();  
  94.     src=cvCreateImage(cvGetSize(workImg),IPL_DEPTH_64F,workImg->nChannels);  //源图像  
  95.     imageClone(workImg,&src);  
  96.     cvFlip(src);  
  97.   
  98.     Fourier = cvCreateImage(cvGetSize(src),IPL_DEPTH_64F,2);  
  99.     dst = cvCreateImage(cvGetSize(src),IPL_DEPTH_64F,2);  
  100.     ImageRe = cvCreateImage(cvGetSize(src),IPL_DEPTH_64F,1);  
  101.     ImageIm = cvCreateImage(cvGetSize(src),IPL_DEPTH_64F,1);  
  102.     Image = cvCreateImage(cvGetSize(src),src->depth,src->nChannels);  
  103.     ImageDst = cvCreateImage(cvGetSize(src),src->depth,src->nChannels);  
  104.     fft2(src,Fourier);                  //傅里叶变换  
  105.     fft2shift(Fourier, Image);          //中心化  
  106.     cvDFT(Fourier,dst,CV_DXT_INV_SCALE);//实现傅里叶逆变换,并对结果进行缩放  
  107.     cvSplit(dst,ImageRe,ImageIm,0,0);  
  108.   
  109.     cvNamedWindow("源图像",0);  
  110.     cvShowImage("源图像",src);               
  111.     //对数组每个元素平方并存储在第二个参数中  
  112.     cvPow(ImageRe,ImageRe,2);                 
  113.     cvPow(ImageIm,ImageIm,2);  
  114.     cvAdd(ImageRe,ImageIm,ImageRe,NULL);  
  115.     cvPow(ImageRe,ImageRe,0.5);  
  116.     cvMinMaxLoc(ImageRe,&m,&M,NULL,NULL);  
  117.     scale = 255/(M - m);  
  118.     shift = -m * scale;  
  119.     //将shift加在ImageRe各元素按比例缩放的结果上,存储为ImageDst  
  120.     cvConvertScale(ImageRe,ImageDst,scale,shift);  
  121.   
  122.     cvNamedWindow("傅里叶谱",0);  
  123.     cvShowImage("傅里叶谱",Image);  
  124.     cvNamedWindow("傅里叶逆变换",0);  
  125.     cvShowImage("傅里叶逆变换",ImageDst);  
  126.     //释放图像  
  127.     cvWaitKey(10000);  
  128.     cvReleaseImage(&src);  
  129.     cvReleaseImage(&Image);  
  130.     cvReleaseImage(&ImageIm);  
  131.     cvReleaseImage(&ImageRe);  
  132.     cvReleaseImage(&Fourier);  
  133.     cvReleaseImage(&dst);  
  134.     cvReleaseImage(&ImageDst);  
  135.     Invalidate();  
  136. }  


from: http://blog.csdn.NET/abcjennifer/article/details/7359952

离散傅里叶变换就是将图像从空间域转换到频域,这一转换基本原理为:

任一函数都可以表示成无数个正弦和余弦函数的和的形式,二维图像的傅里叶变换可用公式表示为:

其中,f是空间域,F是频域,转换之后的频域值是复数,因此显示傅里叶变换之后的结果需要使用实物图像加虚数图像或者幅度图像加相位图像的形式。

示例;

[cpp] view plain copy
  1. #include"stdafx.h"  
  2. #include <opencv2/core/utility.hpp>  
  3. #include "opencv2/imgproc.hpp"  
  4. #include "opencv2/imgcodecs.hpp"  
  5. #include "opencv2/highgui.hpp"  
  6. #include"opencv2/core/core.hpp"  
  7. #include <iostream>  
  8.   
  9. using namespace cv;  
  10. using namespace std;  
  11.   
  12. void DFT_Change(Mat &, string);  
  13. // 定义全局变量  
  14.   
  15.   
  16. int main() {  
  17.     system("color 5E");  
  18.     //读取图片  
  19.     Mat image1 = imread("E:\\pictures\\For_Project\\Opencv_Test\\哈尔施特塔\\05.jpg", 0);  
  20.     namedWindow("【原始图像1】", 1);  
  21.     imshow("【原始图像1】", image1);  
  22.   
  23.     Mat image2 = imread("E:\\pictures\\For_Project\\Opencv_Test\\哈尔施特塔\\b4740bd7f5ee276803fdaa95bfe128d6.jpg", 0);  
  24.     namedWindow("【原始图像2】", 1);  
  25.     imshow("【原始图像2】", image2);  
  26.   
  27.     Mat image3 = imread("E:\\pictures\\For_Project\\Opencv_Test\\哈尔施特塔\\cbf4d74061b713bb5da0bdafedbf1178.jpg", 0);  
  28.     namedWindow("【原始图像3】", 1);  
  29.     imshow("【原始图像3】", image3);  
  30.       
  31.     double time1 = static_cast<double>(getTickCount());  
  32.     //进行DFT傅里叶变换操作   
  33.     string windows_name1 = "频谱幅值1";  
  34.     DFT_Change(image1,windows_name1);  
  35.     //计算运行时间并输出    
  36.     time1 = ((double)getTickCount() - time1) / getTickFrequency();  
  37.     cout << "1.该方法运行时间为:" << time1 << "秒" << endl;  
  38.   
  39.     double time2 = static_cast<double>(getTickCount());  
  40.     //进行DFT傅里叶变换操作   
  41.     string windows_name2 = "频谱幅值2";  
  42.     DFT_Change(image2,windows_name2);  
  43.     //计算运行时间并输出    
  44.     time2 = ((double)getTickCount() - time2) / getTickFrequency();  
  45.     cout << "2.该方法运行时间为:" << time2 << "秒" << endl;  
  46.   
  47.     double time3 = static_cast<double>(getTickCount());  
  48.     //进行DFT傅里叶变换操作   
  49.     string windows_name3 = "频谱幅值3";  
  50.     DFT_Change(image3,windows_name3);  
  51.     //计算运行时间并输出    
  52.     time3 = ((double)getTickCount() - time3) / getTickFrequency();  
  53.     cout << "3.该方法运行时间为:" << time3 << "秒" << endl;  
  54.   
  55.     //等待键盘按键‘q’退出  
  56.     while (char(waitKey(1)) != 'q') {}  
  57.     return 0;  
  58. }  
  59.   
  60. void DFT_Change(Mat &image,string windows_name) {  
  61.       
  62.     Mat srcImage = image.clone();  
  63.     // ShowHelpText();  
  64.   
  65.     // 将输入图像延扩到最佳的尺寸,边界用0补充q  
  66.     int m = getOptimalDFTSize(srcImage.rows);  
  67.     int n = getOptimalDFTSize(srcImage.cols);  
  68.   
  69.     //将添加的像素值初始化为0  
  70.     Mat padded;  
  71.     copyMakeBorder(srcImage, padded, 0, m - srcImage.rows, 0, n - srcImage.cols, BORDER_CONSTANT, Scalar::all(0));  
  72.   
  73.     //为傅里叶变换的结果(实部和虚部)分配存储空间  
  74.     //讲planes数组组合成一个多通道的数组complexI  
  75.     Mat planes[] = { Mat_<float>(padded),Mat::zeros(padded.size(),CV_32F) };  
  76.     Mat complexI;  
  77.     merge(planes, 2, complexI);  
  78.   
  79.     //就地进行离散傅里叶变换  
  80.     dft(complexI, complexI);  
  81.   
  82.     //将复数转换为幅值  
  83.     //讲多通道的数组complexI分离成几个单通道数组  
  84.     split(complexI, planes);  
  85.     magnitude(planes[0], planes[1], planes[0]);  
  86.     Mat magnitudeImage = planes[0];  
  87.   
  88.     //进行对数尺度缩放  
  89.     magnitudeImage += Scalar::all(1);  
  90.     log(magnitudeImage, magnitudeImage);  
  91.   
  92.     //剪切和重分布幅度图像象限  
  93.     //若有奇数或奇数行,进行频谱裁剪  
  94.     magnitudeImage = magnitudeImage(Rect(0, 0, magnitudeImage.cols&-2, magnitudeImage.rows&-2));  
  95.   
  96.     //重新排列傅里叶图像中的象限,使得原点在图像中心  
  97.     int cx = magnitudeImage.cols / 2;  
  98.     int cy = magnitudeImage.rows / 2;  
  99.     //ROI区域的左上  
  100.     Mat q0(magnitudeImage, Rect(0, 0, cx, cy));  
  101.     //ROI区域的右上  
  102.     Mat q1(magnitudeImage, Rect(cx, 0, cx, cy));  
  103.     //ROI区域的左下  
  104.     Mat q2(magnitudeImage, Rect(0, cy, cx, cy));  
  105.     //ROI区域的右下  
  106.     Mat q3(magnitudeImage, Rect(cx, cy, cx, cy));  
  107.   
  108.     //交换象限,左上与右下交换,右上和左下交换  
  109.     Mat tmp1, tmp2;  
  110.     q0.copyTo(tmp1);  
  111.     q3.copyTo(q0);  
  112.     tmp1.copyTo(q3);  
  113.   
  114.     q1.copyTo(tmp2);  
  115.     q2.copyTo(q1);  
  116.     tmp2.copyTo(q2);  
  117.   
  118.     //归一化处理,用0-1之间的浮点值将矩阵变换为可视的图像格式  
  119.     normalize(magnitudeImage, magnitudeImage, 0, 1, NORM_MINMAX);  
  120.   
  121.     imshow(windows_name, magnitudeImage);  
  122. }  


http://www.cnblogs.com/tornadomeet/archive/2012/07/26/2610414.html

原创粉丝点击