Обратная фильтрация в OpenCV - доступ к значениям DFT и умножение матриц DFT - PullRequest
Новая
       12

Обратная фильтрация в OpenCV - доступ к значениям DFT и умножение матриц DFT

4 голосов
/ 29 сентября 2011

Я пытаюсь выполнить обратную и псевдообратную фильтрацию в частотной области.

Однако у меня возникают проблемы с доступом к коэффициентам DFT и умножением матриц DFT впоследствии, так как я получил комплексные числа и, следовательно,фактически две матрицы ...

В основном обратная фильтрация выполняет

F = G / H,

, где F - восстановленное изображение, G - размытое изображение, а H -ядро, которое размыло изображение.

Псевдообратный должен получить доступ к значениям в H, поскольку, если значение близко к 0, его следует заменить, чтобы избежать проблем при восстановлении.Для этого мы должны изменить H так, чтобы:

H (u, v) = 1 / H (u, v), если H (u, v)> порог, и = 0 в противном случае

У меня есть kernel1 (h_1), а также изображения imf (восстановлено) и img (размыто).Вот код: 

  // compute the DFTs of the kernel (DFT_B) and the blurred image (DBF_A)
  cvDFT( dft_A, dft_A, CV_DXT_FORWARD, complexInput1->height );
  cvDFT( dft_B, dft_B, CV_DXT_FORWARD, complexInput2->height );

  // the first type is the inverse fitlering
  if (type == 1) {
      printf("...performing inverse filtering\n");
      // dividing the transforms
      cvDiv(dft_A, dft_B, dft_C, 1);
  }
  // the second type is the pseudo-inverse filtering
  else {
      printf("...prepare kernel for pseudo-inverse filtering\n");
      // will try to access the real values in order to see if value is above a threshold
      cvSplit( dft_B, image_Re1, image_Im1, 0, 0 );
      // pointers to access the data into the real and imaginary matrices
      uchar * dRe1 = (uchar *)image_Re1->imageData;
      uchar * dIm1 = (uchar *)image_Im1->imageData;
      int width  = image_Re1->width;
      int height = image_Re1->height;
      int step = image_Re1->widthStep;
      image_Re2 = cvCreateImage(cvGetSize(image_Re1), IPL_DEPTH_32F, 1);
      image_Im2 = cvCreateImage(cvGetSize(image_Im2), IPL_DEPTH_32F, 1);
      // pointers to access the data into the real and imaginary matrices
      // it will be the resulting pseudo-inverse filter
      uchar * dRe2 = (uchar *)image_Re2->imageData;
      uchar * dIm2 = (uchar *)image_Im2->imageData;

      printf("...building kernel for pseudo-inverse filtering\n");
      for ( i = 0; i < height; i++ )  {
      for ( j = 0; j < width; j++ ) {
                // generate the 1/H(i,j) value
        if (dRe1[i * step + j] > threshold) {
            float realsq = dRe1[i * step + j]*dRe1[i * step + j];
            float imagsq = dIm1[i * step + j]*dIm1[i * step + j];

            dRe2[i * step + j] = dRe1[i * step + j] / (realsq + imagsq);
            dIm2[i * step + j] = -1 * (dIm1[i * step + j] / (realsq + imagsq));
        }
        else {
            dRe2[i * step + j] = 0;
            dIm2[i * step + j] = 0;
        }
      }
      }
      printf("...merging final kernel\n");
      cvMerge(image_Re2, image_Im2, 0, 0, dft_B);
      printf("...performing pseudo-inverse filtering\n");
      cvMulSpectrums(dft_A, dft_B, dft_C, 1);
  }
  printf("...performing IDFT\n");
  cvDFT(dft_C, dft_H, CV_DXT_INV_SCALE, 1);

  printf("...getting size\n");
  cvGetSubRect(dft_H, &tmp3, cvRect(0, 0, img->width, img->height));

  printf("......(%d, %d) - (%d, %d)\n", tmp3.cols, tmp3.rows, restored->width, restored->height);

  cvSplit( &tmp3, image_Re1, image_Im1, 0, 0 );

  cvNamedWindow("re", 0);
  cvShowImage("re", image_Re2);
  cvWaitKey(0);

  printf("...copying final image\n");
  // error is in the line below
  cvCopy(image_Re1, imf, NULL);

У меня есть ошибка в последней строке: --- Ошибка OpenCV: утверждение не выполнено (src.depth () == dst.depth () && src.size ()== dst.size ()) в cvCopy, файл /build/buildd/opencv-2.1.0/src/cxcore/cxcopy.cpp, строка 466

Я знаю, что это связано с размером или глубинойно я не знаю, как контролировать.Во всяком случае, я пытался показать image_Re1, и он пуст ...

Может кто-нибудь пролить свет на это?

1 Ответ

2 голосов
/ 21 октября 2011

Похоже, вы не инициализировали свою картинку imf! cvCopy нужна инициализированная матрица: 

IplImage* imf=   cvCreateImage(cvGetSize(image_Re1), IPL_DEPTH_32F, 1);

сначала, и я думаю, что это сработает.

Кроме того, вы не освобождаете пространство изображения в этом коде (cvReleaseImage(&image))

...