1

C++ インターフェイスを使用して、FFTW 3.3 および OpenCV 2.1 で FFT/IFFT 関数をコーディングしようとしています。古い OpenCV 形式を使用した多くの例を見て、直接変換を行いましたが、何かがうまくいきません。

私の関数の目的は、dft のデフォルトの OpenCV 関数が行うように、FFT の実部と虚部を含む Mat オブジェクトを返すことです。これが関数のコードです。im_data を data_in にコピーする行で、プログラムがメモリの問題でブロックされます。

誰かが私が間違っていることを知っていますか? ありがとうございました

Mat fft_sr(Mat& I)
{

double          *im_data;
double          *realP_data;
double          *imP_data;

fftw_complex    *data_in;
fftw_complex    *fft;      

fftw_plan       plan_f;

int width     = I.cols;
int height    = I.rows;
int step      = I.step;

int             i, j, k;

Mat realP=Mat::zeros(height,width,CV_64F); // Real Part FFT
Mat imP=Mat::zeros(height,width,CV_64F); // Imaginary Part FFT


im_data = ( double* ) I.data;
realP_data = ( double* ) realP.data;
imP_data = ( double* ) imP.data;


data_in = ( fftw_complex* )fftw_malloc( sizeof( fftw_complex ) * width * height );
fft     = ( fftw_complex* )fftw_malloc( sizeof( fftw_complex ) * width * height );

// Problem Here
for( i = 0, k = 0 ; i < height ; i++ ) {
    for( j = 0 ; j < width ; j++ ) {
        data_in[k][0] = ( double )im_data[i * step + j];
        data_in[k][1] = ( double )0.0;
        k++;
    }
}


plan_f = fftw_plan_dft_2d( height, width, data_in, fft,  FFTW_FORWARD,  FFTW_ESTIMATE );


fftw_execute( plan_f );

// Copy real and imaginary data
for( i = 0, k = 0 ; i < height ; i++ ) {
    for( j = 0 ; j < width ; j++ ) {
        realP_data[i * step + j] = ( double )fft[k][0];
        imP_data[i * step + j] = ( double )fft[k][1];
        k++;

    }
}



Mat fft_I(I.size(),CV_64FC2);
Mat fftplanes[] = {Mat_<double>(realP), Mat_<double>(imP)};
merge(fftplanes, 2, fft_I);

fftw_destroy_plan(plan_f);
fftw_free(data_in);
fftw_free(fft);
return fft_I;   
}
4

3 に答える 3

3

使い方がstep間違っています。に索引付けするためのものMat::dataです。に割り当てるときにすでに にキャストMat::dataしているため、 「通常」にインデックスを付けることができます。double*im_dataim_data

data_in[k][0] = im_data[i * width + j];

stepインデックスを作成する正しい方法を使用する場合:

data_in[k][0] = ( double )I.data[i * step + j];

アップデート：

行ごとに画像にアクセスしてみてください。そうすれば、高速アクセスを利用しながら、ストライド/ステップの問題に遭遇することを回避できます。

for (int i = 0; i < I.rows; i++)
{
    double* row = I.ptr<double>(i);

    for (int j = 0; j < I.cols; j++)
    {
        // Do something with the current pixel.
        double someValue = row[j];
    }
}
于 2012-11-26T13:04:56.077 に答える