本文實例講述了C++實現二維圖形的傅里葉變換的方法。有一定的借鑒價值。分享給大家供大家參考。
具體代碼如下:
// Fourier.cpp : Defines the entry point for the console application.//#include "stdafx.h"#include "stdio.h"#include "math.h"#include <cv.h>#include <highgui.h>#include "cxcore.h"int main(int argc, char* argv[]){ IplImage *img; IplImage *simg; CvMat *mat_R; CvMat *mat_I; CvMat *mat_SRC; CvMat *mat_Row; CvMat *mat_Col; CvMat *dst; CvMat *dst_R; CvMat *dst_I; CvMat *dst_Row; CvMat *dst_Col; int i,j,k; double temp; int height,width,step,channels; //載入一幅圖片 img=cvLoadImage("c://1.bmp",0); //mat_R初始化 mat_R=cvCreateMat(img->height,img->width,CV_64FC1); //mat_I初始化 mat_I=cvCreateMat(img->height,img->width,CV_64FC1); //mat_SRC初始化 mat_SRC=cvCreateMat(img->height,img->width,CV_64FC2); //將圖片數據存入mat_R(實部) cvConvert(img,mat_R); //將虛部初始化為零 cvZero(mat_I); //合并實部、虛部 cvMerge(mat_R,mat_I,NULL,NULL,mat_SRC); //創建雙通道double類型數組 dst=cvCreateMat(img->height,img->width,CV_64FC2); dst_R=cvCreateMat(img->height,img->width,CV_64FC1); dst_I=cvCreateMat(img->height,img->width,CV_64FC1); //為循環變量賦值 height=img->height; width=img->width; channels=2; step=channels*width; //局部變量,值為正一或負一 int check; //將輸入數據乘以(-1)^(i+j),用于中心化 for(j=0;j<height;j++) { for(i=0;i<width;i++) { check=(i+j)%2>0?1:-1; for(k=0;k<channels;k++) { mat_SRC->data.db[j*step+i*channels+k]=check*mat_SRC->data.db[j*step+i*channels+k]; } } } //創建一個mat用于臨時存儲一行數據 CvMat mat_Header=cvMat(4,4,CV_64FC2); mat_Row=cvCreateMat(1,width,CV_64FC2); mat_Col=cvCreateMat(1,height,CV_64FC2); //創建一個dst用于臨時存儲一行數據 dst_Row=cvCreateMat(1,width,CV_64FC2); dst_Col=cvCreateMat(height,1,CV_64FC2); //為循環變量賦值 height=img->height; width=img->width; channels=2; step=channels*width; //行的傅里葉變換 for(j=0;j<height;j++) { //取得第j行數據 mat_Row=cvGetRow(mat_SRC,&mat_Header,j); //正向傅里葉變換 cvDFT(mat_Row,dst_Row,CV_DXT_FORWARD); //執行循環,賦值到dst for(i=0;i<width;i++) { for(k=0;k<channels;k++) { dst->data.db[j*step+i*channels+k]=dst_Row->data.db[i*channels+k]; } } } //列的傅里葉變換 for(i=0;i<width;i++) { //取得第i列 mat_Col=cvGetCol(dst,&mat_Header,i); //正向傅里葉變換 cvDFT(mat_Col,dst_Col,CV_DXT_FORWARD); //執行循環,賦值到dst for(j=0;j<height;j++) { for(k=0;k<channels;k++) { dst->data.db[j*step+i*channels+k]=dst_Col->data.db[j*channels+k]; } } } //分成兩個矩陣 cvSplit(dst,dst_R,dst_I,NULL,NULL); //創建臨時指針指向dst_R,dst_I double *pR,*pI; pR=(double *)dst_R->data.ptr; pI=(double *)dst_I->data.ptr; //創建一張用于顯示的圖像 simg=cvCreateImage(cvGetSize(img),8,1); //為循環變量賦值 height=simg->height; width=simg->width; channels=1; step=channels*width; for(j=0;j<height;j++) { for(i=0;i<width;i++) { for(k=0;k<channels;k++) { temp=pR[j*step+i*channels+k]*pR[j*step+i*channels+k]+pI[j*step+i*channels+k]*pI[j*step+i*channels+k]; temp=temp/(height*width); simg->imageData[j*step+i*channels+k]=sqrt(temp); } } } cvNamedWindow("Mar",CV_WINDOW_AUTOSIZE); cvShowImage("Mar",simg); cvWaitKey(0); cvReleaseMat(&mat_R); cvReleaseMat(&mat_I); cvReleaseMat(&mat_SRC); //cvReleaseMat(&mat_Row);//這里無法正常釋放,有待解決 //cvReleaseMat(&mat_Col); cvReleaseMat(&dst); cvReleaseMat(&dst_R); cvReleaseMat(&dst_I); cvReleaseImage(&img); cvReleaseImage(&simg); return 0;}
感興趣的朋友可以調試運行一下本文實例,程序美中不足的是會有內存泄漏,主要是mat_Row,mat_Col,dst_Row,dst_Col,有能力的讀者可以對此進行修改與完善。相信會有新的收獲。
新聞熱點
疑難解答
圖片精選