开通博客赚积分
发布资源赚积分
分类

源码开发语言/平台
当前位置：首页 > 源码/资料 > 图形图像 > 波变换 > 查看源码
gabor.cpp
资源名称：gabor.rar　[点击查看]
上传用户：soukeisyuu
上传日期：2022-07-03
资源大小：5943k
文件大小：5k
源码类别：

波变换

开发平台：

Visual C++

gabor.cpp：源码内容
  1. #include "gabor.h"
  3. CvGabor::CvGabor()
  4. {
  5. }
  6. CvGabor::~CvGabor()
  7. {
  8. }
  9. void CvGabor::initGabor(int fre,int ang,int size,float sigma)
  10. {
  11.     s_fre = fre;
  12.     s_ang = ang;
  13.     aperture_size  = size;
  14.     gaussian_sigma = sigma;
  15.     freq = 0.5*PI/pow(sqrt(2.0),s_fre);
  16.     angle = PI*s_ang/8;
  17. }
  19. void CvGabor::doGabor(IplImage *src,IplImage *dst)
  20. {
  21.     CvMat *floatImg;
  22.     CvMat *gaborReal,*gaborImage;
  23.     CvMat *convReal,*convImage;
  24.     int gaborSize = aperture_size;
  25.     int halfSize  = (gaborSize-1)/2;
  26.     float sigma = gaussian_sigma;
  28.     floatImg =  cvCreateMat(src->height,src->width,CV_32FC1);
  29.     gaborReal = cvCreateMat(gaborSize,gaborSize,CV_32FC1);
  30.     gaborImage = cvCreateMat(gaborSize,gaborSize,CV_32FC1);
  31.     convReal = cvCreateMat(src->height+gaborSize-1,src->width+gaborSize+1,CV_32FC1);
  32.     convImage = cvCreateMat(src->height+gaborSize-1,src->width+gaborSize+1,CV_32FC1);
  34.     char* src_data = src->imageData;
  35.     int src_step = src->widthStep/sizeof(char);
  36.     float* floatImg_data = floatImg->data.fl;
  37.     int floatImg_step = floatImg->step/sizeof(float);
  38.     for(int i=0;i<src->height;i++)
  39.         for(int j=0;j<src->width;j++)
  40.             floatImg_data[i*floatImg_step+j] = (float)src_data[i*src_step+j];
  42.     float gaussPart,realPart,imagePart;
  44.     float* gaborReal_data = gaborReal->data.fl;
  45.     int gaborReal_step = gaborReal->step/sizeof(float);
  46.     float* gaborImage_data = gaborImage->data.fl;
  47.     int gaborImage_step = gaborImage->step/sizeof(float);
  48.     for(int i=0;i<gaborSize;i++)
  49.         for(int j=0;j<gaborSize;j++)
  50.         {
  51.             realPart = cos( freq*cos(angle)*(j-halfSize) + freq*sin(angle)*(i-halfSize) )  - exp( -0.5*sigma*sigma );
  52.             imagePart = sin( freq*cos(angle)*(j-halfSize) + freq*sin(angle)*(i-halfSize) );
  53.             gaussPart = (freq*freq)/(sigma*sigma)*exp( -0.5*(freq*freq)/(sigma*sigma)*((j-halfSize)*(j-halfSize)+(i-halfSize)*(i-halfSize)) );
  54.             gaborReal_data[i*gaborReal_step+j] = gaussPart*realPart;
  55.             gaborImage_data[i*gaborImage_step+j] = gaussPart*imagePart;
  56.         }
  58.     doConv(floatImg,gaborReal,convReal);
  59.     doConv(floatImg,gaborImage,convImage);
  60.     cvPow(convReal,convReal,2.0);
  61.     cvPow(convImage,convImage,2.0);
  63.     char* dst_data = dst->imageData;
  64.     int dst_step = dst->widthStep/sizeof(char);
  65.     float* convReal_data = convReal->data.fl;
  66.     int convReal_step = convReal->step/sizeof(float);
  67.     float* convImage_data = convImage->data.fl;
  68.     int convImage_step = convImage->step/sizeof(float);
  69.     for(int i=0;i<dst->height;i++)
  70.         for(int j=0;j<dst->width;j++)
  71.         {
  72.             dst_data[i*dst_step+j] = min( 255, (int)sqrt(convReal_data[i*convReal_step+j]+convImage_data[i*convImage_step+j]) );
  73.         }
  75.     cvReleaseMat(&floatImg);
  76.     cvReleaseMat(&gaborReal);
  77.     cvReleaseMat(&gaborImage);
  78.     cvReleaseMat(&convReal);
  79.     cvReleaseMat(&convImage);
  80. }
  82. void CvGabor::doConv(CvMat *A,CvMat *B,CvMat* conv)
  83. {
  84.     /*
  85.     float a[] = {1,2,3,4};
  86.     float b[] = {2,3,4,5};
  87.     CvMat *A = new CvMat;
  88.     CvMat *B = new CvMat;
  89.     cvInitMatHeader(A,2,2,CV_32F,a);
  90.     cvInitMatHeader(B,2,2,CV_32F,b);
  91.     */
  93.     /*
  94.     CvMat* conv = cvCreateMat( A->rows + B->rows - 1, A->cols + B->cols - 1,A->type );
  95.     */
  97.     int dftgM = cvGetOptimalDFTSize( A->rows + B->rows - 1 );
  98.     int dftgN = cvGetOptimalDFTSize( A->cols + B->cols - 1 );
  99.     CvMat* dftgA = cvCreateMat( dftgM, dftgN, A->type );
  100.     CvMat* dftgB = cvCreateMat( dftgM, dftgN, B->type );
  101.     CvMat tmp;
  103.     cvGetSubRect( dftgA, &tmp, cvRect(0,0,A->cols,A->rows));
  104.     cvCopy( A, &tmp );
  105.     cvGetSubRect( dftgA, &tmp, cvRect(A->cols,0,dftgA->cols - A->cols,A->rows));
  106.     cvZero( &tmp );
  107.     cvDFT( dftgA, dftgA, CV_DXT_FORWARD, A->rows );
  109.     cvGetSubRect( dftgB, &tmp, cvRect(0,0,B->cols,B->rows));
  110.     cvCopy( B, &tmp );
  111.     cvGetSubRect( dftgB, &tmp, cvRect(B->cols,0,dftgB->cols - B->cols,B->rows));
  112.     cvZero( &tmp );
  113.     cvDFT( dftgB, dftgB, CV_DXT_FORWARD, B->rows );
  115.     cvMulSpectrums( dftgA, dftgB, dftgA, 0);
  116.     cvDFT( dftgA, dftgA, CV_DXT_INV_SCALE, conv->rows );
  118.     cvGetSubRect( dftgA, &tmp, cvRect(0,0,conv->cols,conv->rows) );
  119.     cvCopy( &tmp, conv );
  121.     /*
  122.    for(int i=0;i<2;i++)
  123.     {
  124.         for(int j=0;j<2;j++)
  125.         {
  126.             printf("%ft",cvmGet(A,i,j));
  127.         }
  128.         printf("n");
  129.     }
  130.     printf("n");
  132.     for(int i=0;i<2;i++)
  133.     {
  134.         for(int j=0;j<2;j++)
  135.         {
  136.             printf("%ft",cvmGet(B,i,j));
  137.         }
  138.         printf("n");
  139.     }
  140.     printf("n");
  141.     */
  143.     /*
  144.     for(int i=0;i<3;i++)
  145.     {
  146.         for(int j=0;j<3;j++)
  147.         {
  148.             printf("%ft",cvmGet(conv,i,j));
  149.         }
  150.         printf("n");
  151.     }
  152.     printf("n");
  153.     */
  155.     cvReleaseMat(&dftgA);
  156.     cvReleaseMat(&dftgB);
  157. }