iplImage avec la bibliothèque FFTW
Bonjour,
Vu les bonnes performances de la bibliothèque FFTW, je l'ai installée avec openCV 2.3.1 pour calculer la fonction d'auto-corrélation d'une image avec ce code suivant:
Code:
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
|
/*
* get phase correlation from two images and save result to the third image
*/
void Segment_R::phase_correlation(const IplImage *ref,const IplImage *tpl, IplImage *poc )
{
int i, j, k;
double tmp;
/* get image properties */
int width = ref->width;
int height = ref->height;
int step = ref->widthStep;
int fft_size = width * height;
/* setup pointers to images */
uchar *ref_data = ( uchar* ) ref->imageData;
uchar *tpl_data = ( uchar* ) tpl->imageData;
double *poc_data = ( double* )poc->imageData;
/* allocate FFTW input and output arrays */
fftw_complex *img1 = ( fftw_complex* )fftw_malloc( sizeof( fftw_complex ) * width * height );
fftw_complex *img2 = ( fftw_complex* )fftw_malloc( sizeof( fftw_complex ) * width * height );
fftw_complex *res = ( fftw_complex* )fftw_malloc( sizeof( fftw_complex ) * width * height );
/* setup FFTW plans */
fftw_plan fft_img1 = fftw_plan_dft_1d( width * height, img1, img1, FFTW_FORWARD, FFTW_ESTIMATE );
fftw_plan fft_img2 = fftw_plan_dft_1d( width * height, img2, img2, FFTW_FORWARD, FFTW_ESTIMATE );
fftw_plan ifft_res = fftw_plan_dft_1d( width * height, res, res, FFTW_BACKWARD, FFTW_ESTIMATE );
/* load images' data to FFTW input */
for( i = 0, k = 0 ; i < height ; i++ ) {
for( j = 0 ; j < width ; j++, k++ ) {
img1[k][0] = ( double )ref_data[i * step + j];
img1[k][1] = 0.0;
img2[k][0] = ( double )tpl_data[i * step + j];
img2[k][1] = 0.0;
}
}
/* obtain the FFT of img1 */
fftw_execute( fft_img1 );
/* obtain the FFT of img2 */
fftw_execute( fft_img2 );
/* obtain the cross power spectrum */
for( i = 0; i < fft_size ; i++ ) {
res[i][0] = ( img2[i][0] * img1[i][0] ) - ( img2[i][1] * ( -img1[i][1] ) );
res[i][1] = ( img2[i][0] * ( -img1[i][1] ) ) + ( img2[i][1] * img1[i][0] );
tmp = sqrt( pow( res[i][0], 2.0 ) + pow( res[i][1], 2.0 ) );
res[i][0] /= tmp;
res[i][1] /= tmp;
}
/* obtain the phase correlation array */
fftw_execute(ifft_res);
/* normalize and copy to result image */
for( i = 0 ; i < fft_size ; i++ ) {
poc_data[i] = res[i][0] / ( double )fft_size;
//qDebug()<< "i: " << i << "poc_data: " << poc_data[i];
}
/* deallocate FFTW arrays and plans */
fftw_destroy_plan( fft_img1 );
fftw_destroy_plan( fft_img2 );
fftw_destroy_plan( ifft_res );
fftw_free( img1 );
fftw_free( img2 );
fftw_free( res );
} |
Et, j'appelle par la suite la fonction phase_correlation sur une image de grande taille et la fonction marche très bien mais lorsque je fais un crop ou une région d’intérêt (ROI) de taille 32*32, ça ne marche pas et il m'envoie des valeurs nulles avec le code suivant:
Code:
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
|
cv::Mat mat= cv::Mat((IplImage*)iplImage);
int posxroi = 50;
int posyroi = 50;
int heightroi = 32;
int widthroi = 32;
Rect r = Rect(posxroi, posyroi, widthroi, heightroi);
Rect rectinter = r & Rect(0,0,iplImage->width,iplImage->height);
if (rectinter!= Rect())
{
cv::Mat mat_roi= mat(rectinter);
IplImage* iplImage_ROI;// = cvCreateImage(cvSize(widthroi, heightroi), iplImage->depth, iplImage->nChannels);
IplImage iplImg = (IplImage)mat_roi;
iplImage_ROI= &iplImg;
IplImage *poc;
/* create a new image, to store phase correlation result */
poc = cvCreateImage( cvSize( iplImage->width, iplImage->height ), IPL_DEPTH_64F, 1 );
/* get phase correlation of input images */
phase_correlation( iplImage, iplImage, poc );
}
else
return
QString(); |
Je suis suis sûre que le problème c'est la fftw qui ne peut être calculée sur des image de petite taille (32*32 pixels).
Mais je ne sais pas comment pourrais je résoudre le problème, et j'ai modifié les options de FFTW selon le site (http://www.fftw.org/faq/section3.html#allzero): FFTW_MEASURE, FFTW_PATIENT et FFTW_EXAUSTIVE mais le même problème persiste.
Quelqu'un pourra m'aider?
Merci.
Cordialement,
Maroua
