gabor函数的c++实现

Gabor2.h

#ifndef CVGABOR_H
#define CVGABOR_H

#include <iostream>


#include <cv.h>
#include <highgui.h>

#define PI 3.14159265
#define CV_GABOR_REAL 1
#define CV_GABOR_IMAG 2
#define CV_GABOR_MAG   3
#define CV_GABOR_PHASE 4

class CvGabor{
public:
CvGabor();
~CvGabor();


CvGabor(int iMu, int iNu, double dSigma);
CvGabor(int iMu, int iNu, double dSigma, double dF);
CvGabor(double dPhi, int iNu);
CvGabor(double dPhi, int iNu, double dSigma);
CvGabor(double dPhi, int iNu, double dSigma, double dF);
bool IsInit();
long mask_width();
IplImage* get_image(int Type);
bool IsKernelCreate();
long get_mask_width();
void Init(int iMu, int iNu, double dSigma, double dF);
void Init(double dPhi, int iNu, double dSigma, double dF);
void output_file(const char *filename, int Type);
CvMat* get_matrix(int Type);
void show(int Type);
void conv_img(IplImage *src, IplImage *dst, int Type);
CvGabor(int iMu, int iNu);
void normalize( const CvArr* src, CvArr* dst, double a, double b, int norm_type, const CvArr* mask );
void conv_img_a(IplImage *src, IplImage *dst, int Type);

protected:
double Sigma;
double F;
double Kmax;
double K;
double Phi;
bool bInitialised;
bool bKernel;
long Width;
CvMat *Imag;
CvMat *Real;

private:
void creat_kernel();


};

#endif


Gabor2.cpp


#include "Gabor2.h"


#define   round(x)   ((int)(x   +   0.5))    

CvGabor::CvGabor()
{
}


CvGabor::~CvGabor()
{
cvReleaseMat( &Real );
cvReleaseMat( &Imag );
}




CvGabor::CvGabor(int iMu, int iNu, double dSigma)
    F = sqrt(2.0);
    Init(iMu, iNu, dSigma, F);
}


  CvGabor::CvGabor(int iMu, int iNu, double dSigma, double dF)
{

    Init(iMu, iNu, dSigma, dF);
     
}


  CvGabor::CvGabor(double dPhi, int iNu)
{
    Sigma = 2*PI;
    F = sqrt(2.0);
    Init(dPhi, iNu, Sigma, F);
}


  CvGabor::CvGabor(double dPhi, int iNu, double dSigma)
{

    F = sqrt(2.0);
    Init(dPhi, iNu, dSigma, F);
}


  CvGabor::CvGabor(double dPhi, int iNu, double dSigma, double dF)
{

    Init(dPhi, iNu, dSigma,dF);
}

bool CvGabor::IsInit()
{

    return bInitialised;
}

long CvGabor::mask_width()
{

    long lWidth;
    if (IsInit() == false)   {
        perror ("Error: The Object has not been initilised in mask_width()!\n");
        return 0;
    }
    else {
        //determine the width of Mask
      double dModSigma = Sigma/K;
      double dWidth = round(dModSigma*6 + 1);

      //test whether dWidth is an odd.
      if (fmod(dWidth, 2.0)==0.0) dWidth++;
      lWidth = (long)dWidth;

      return lWidth;
    }
}


void CvGabor::creat_kernel()
{
     
    if (IsInit() == false) {perror("Error: The Object has not been initilised in creat_kernel()!\n");}
    else {
      CvMat *mReal, *mImag;
      mReal = cvCreateMat( Width, Width, CV_32FC1);
      mImag = cvCreateMat( Width, Width, CV_32FC1);
       
       
      int x, y;
      double dReal;
      double dImag;
      double dTemp1, dTemp2, dTemp3;

      for (int i = 0; i < Width; i++)
      {
          for (int j = 0; j < Width; j++)
          {
              x = i-(Width-1)/2;
              y = j-(Width-1)/2;
              dTemp1 = (pow(K,2)/pow(Sigma,2))*exp(-(pow((double)x,2)+pow((double)y,2))*pow(K,2)/(2*pow(Sigma,2)));
              dTemp2 = cos(K*cos(Phi)*x + K*sin(Phi)*y) - exp(-(pow(Sigma,2)/2));
              dTemp3 = sin(K*cos(Phi)*x + K*sin(Phi)*y);
              dReal = dTemp1*dTemp2;
              dImag = dTemp1*dTemp3; 
              //gan_mat_set_el(pmReal, i, j, dReal);
      //cvmSet( (CvMat*)mReal, i, j, dReal );
cvSetReal2D((CvMat*)mReal, i, j, dReal );
              //gan_mat_set_el(pmImag, i, j, dImag);
              //cvmSet( (CvMat*)mImag, i, j, dImag );
cvSetReal2D((CvMat*)mImag, i, j, dImag );

         
        }
       
        bKernel = true;
        cvCopy(mReal, Real, NULL);
        cvCopy(mImag, Imag, NULL);
      //printf("A %d x %d Gabor kernel with %f PI in arc is created.\n", Width, Width, Phi/PI);
        cvReleaseMat( &mReal );
        cvReleaseMat( &mImag );
      }
}


IplImage* CvGabor::get_image(int Type)
{

    if(IsKernelCreate() == false)
   
      perror("Error: the Gabor kernel has not been created in get_image()!\n");
      return NULL;
    }
    else
    {  
    IplImage* pImage;
    IplImage *newimage;
    newimage = cvCreateImage(cvSize(Width,Width), IPL_DEPTH_8U, 1 );
    //printf("Width is %d.\n",(int)Width);
    //printf("Sigma is %f.\n", Sigma);
    //printf("F is %f.\n", F);
    //printf("Phi is %f.\n", Phi);
     
    //pImage = gan_image_alloc_gl_d(Width, Width);
    pImage = cvCreateImage( cvSize(Width,Width), IPL_DEPTH_32F, 1 );
     
     
    CvMat* kernel = cvCreateMat(Width, Width, CV_32FC1);
    double ve;
    CvScalar S;
    CvSize size = cvGetSize( kernel );
    int rows = size.height;
    int cols = size.width;
    switch(Type)
    {
        case 1:   //Real

            cvCopy( (CvMat*)Real, (CvMat*)kernel, NULL );
            //pImage = cvGetImage( (CvMat*)kernel, pImageGL );
            for (int i = 0; i < rows; i++)
      {
              for (int j = 0; j < cols; j++)
              {
                    ve = cvGetReal2D((CvMat*)kernel, i, j);
                    cvSetReal2D( (IplImage*)pImage, j, i, ve );
              }
            }
            break;
        case 2:   //Imag
            cvCopy( (CvMat*)Imag, (CvMat*)kernel, NULL );
            //pImage = cvGetImage( (CvMat*)kernel, pImageGL );
            for (int i = 0; i < rows; i++)
      {
              for (int j = 0; j < cols; j++)
              {
                    ve = cvGetReal2D((CvMat*)kernel, i, j);
                    cvSetReal2D( (IplImage*)pImage, j, i, ve );
              }
            }
            break; 
        case 3:   //Magnitude
            ///@todo  
            break;
        case 4:   //Phase
          ///@todo
            break;
    }
   
    cvNormalize((IplImage*)pImage, (IplImage*)pImage, 0, 255, CV_MINMAX, NULL );


    cvConvertScaleAbs( (IplImage*)pImage, (IplImage*)newimage, 1, 0 );

    cvReleaseMat(&kernel);

    cvReleaseImage(&pImage);

    return newimage;
    }
}


bool CvGabor::IsKernelCreate()
{

    return bKernel;
}


long CvGabor::get_mask_width()
{
  return Width;
}


void CvGabor::Init(int iMu, int iNu, double dSigma, double dF)
{
  //Initilise the parameters 
    bInitialised = false;
    bKernel = false;

    Sigma = dSigma;
    F = dF;
     
    Kmax = PI/2;
     
    // Absolute value of K
    K = Kmax / pow(F, (double)iNu);
    Phi = PI*iMu/8;
    bInitialised = true;
    Width = mask_width();
    Real = cvCreateMat( Width, Width, CV_32FC1);
    Imag = cvCreateMat( Width, Width, CV_32FC1);
    creat_kernel();  
}

void CvGabor::Init(double dPhi, int iNu, double dSigma, double dF)
{

    bInitialised = false;
    bKernel = false;
    Sigma = dSigma;
    F = dF;
     
    Kmax = PI/2;
     
    // Absolute value of K
    K = Kmax / pow(F, (double)iNu);
    Phi = dPhi;
    bInitialised = true;
    Width = mask_width();
    Real = cvCreateMat( Width, Width, CV_32FC1);
    Imag = cvCreateMat( Width, Width, CV_32FC1);
    creat_kernel();  
}



CvMat* CvGabor::get_matrix(int Type)
{
    if (!IsKernelCreate()) {perror("Error: the gabor kernel has not been created!\n"); return NULL;}
    switch (Type)
    {
      case CV_GABOR_REAL:
        return Real;
        break;
      case CV_GABOR_IMAG:
        return Imag;
        break;
      case CV_GABOR_MAG:
        return NULL;
        break;
      case CV_GABOR_PHASE:
        return NULL;
        break;
    }
}




void CvGabor::output_file(const char *filename, int Type)
{
  IplImage *pImage;
  pImage = get_image(Type);
  if(pImage != NULL)
  {
    if( cvSaveImage(filename, pImage )) printf("%s has been written successfully!\n", filename);
    else printf("Error: writting %s has failed!\n", filename);
  }
  else 
    perror("Error: the image is empty in output_file()!\n"); 

  cvReleaseImage(&pImage);
}






void CvGabor::show(int Type)
{
    if(!IsInit()) {
        perror("Error: the gabor kernel has not been created!\n");
    }
    else {
      //     IplImage *pImage;
      //pImage = get_image(Type);
      //cvNamedWindow("Testing",1);
      //cvShowImage("Testing",pImage);
      //cvWaitKey(0);
      //cvDestroyWindow("Testing");
      //cvReleaseImage(&pImage);
    }

}




void CvGabor::conv_img_a(IplImage *src, IplImage *dst, int Type)
{
    double ve, re,im;
   
    int width = src->width;
    int height = src->height;
    CvMat *mat = cvCreateMat(src->width, src->height, CV_32FC1);
     
    for (int i = 0; i < width; i++)
    {
        for (int j = 0; j < height; j++)
        {
              ve = cvGetReal2D((IplImage*)src, j, i);
              cvSetReal2D( (CvMat*)mat, i, j, ve );
        }
    }

    CvMat *rmat = cvCreateMat(width, height, CV_32FC1);
    CvMat *imat = cvCreateMat(width, height, CV_32FC1);

    CvMat *kernel = cvCreateMat( Width, Width, CV_32FC1 );

    switch (Type)
    {
      case CV_GABOR_REAL:
        cvCopy( (CvMat*)Real, (CvMat*)kernel, NULL );
        cvFilter2D( (CvMat*)mat, (CvMat*)mat, (CvMat*)kernel, cvPoint( (Width-1)/2, (Width-1)/2));
        break;
      case CV_GABOR_IMAG:
        cvCopy( (CvMat*)Imag, (CvMat*)kernel, NULL );
        cvFilter2D( (CvMat*)mat, (CvMat*)mat, (CvMat*)kernel, cvPoint( (Width-1)/2, (Width-1)/2));
        break;
      case CV_GABOR_MAG:
       
        cvCopy( (CvMat*)Real, (CvMat*)kernel, NULL );
        cvFilter2D( (CvMat*)mat, (CvMat*)rmat, (CvMat*)kernel, cvPoint( (Width-1)/2, (Width-1)/2));
       
        cvCopy( (CvMat*)Imag, (CvMat*)kernel, NULL );
        cvFilter2D( (CvMat*)mat, (CvMat*)imat, (CvMat*)kernel, cvPoint( (Width-1)/2, (Width-1)/2));
       
        for (int i = 0; i < width; i++)
        {
            for (int j = 0; j < height; j++)
            {
                re = cvGetReal2D((CvMat*)rmat, i, j);
                im = cvGetReal2D((CvMat*)imat, i, j);
                ve = sqrt(re*re + im*im);
                cvSetReal2D( (CvMat*)mat, i, j, ve );
            }
        }        
        break;
      case CV_GABOR_PHASE:
        break;
    }
     
    if (dst->depth == IPL_DEPTH_8U)
    {
        cvNormalize((CvMat*)mat, (CvMat*)mat, 0, 255, CV_MINMAX, NULL);
    for (int i = 0; i < width; i++)
    {
            for (int j = 0; j < height; j++)
            {
                ve = cvGetReal2D((CvMat*)mat, i, j);
                ve = cvRound(ve);
                cvSetReal2D( (IplImage*)dst, j, i, ve );
            }
        }
      }

      if (dst->depth == IPL_DEPTH_32F)
      {
          for (int i = 0; i < width; i++)
    {
            for (int j = 0; j < height; j++)
            {
                ve = cvGetReal2D((CvMat*)mat, i, j);
                cvSetReal2D( (IplImage*)dst, j, i, ve );
            }
          }
     

    cvReleaseMat(&kernel);
    cvReleaseMat(&imat);
    cvReleaseMat(&rmat);
    cvReleaseMat(&mat);
}


  CvGabor::CvGabor(int iMu, int iNu)
{
  double dSigma = 2*PI; 
  F = sqrt(2.0);
  Init(iMu, iNu, dSigma, F);
}


void CvGabor::normalize( const CvArr* src, CvArr* dst, double a, double b, int norm_type, const CvArr* mask )
{
    CvMat* tmp = 0;
//     __BEGIN__;

    double scale, shift;
     
    if( norm_type == CV_MINMAX )
    {
        double smin = 0, smax = 0;
        double dmin = MIN( a, b ), dmax = MAX( a, b );
        cvMinMaxLoc( src, &smin, &smax, 0, 0, mask );
        scale = (dmax - dmin)*(smax - smin > DBL_EPSILON ? 1./(smax - smin) : 0);
        shift = dmin - smin*scale;
    }
    else if( norm_type == CV_L2 || norm_type == CV_L1 || norm_type == CV_C )
    {
        CvMat *s = (CvMat*)src, *d = (CvMat*)dst;

        scale = cvNorm( src, 0, norm_type, mask );
        scale = scale > DBL_EPSILON ? 1./scale : 0.;
        shift = 0;
    }
    else {}

       
     
    if( !mask )
        cvConvertScale( src, dst, scale, shift );
    else
    {
        CvMat stub, *dmat;
 
       
        cvConvertScale( src, tmp, scale, shift );
        cvCopy( tmp, dst, mask );
    }

//     __END__;

    if( tmp )
        cvReleaseMat( &tmp );
}


void CvGabor::conv_img(IplImage *src, IplImage *dst, int Type)
{
  double ve, re,im;
   
  CvMat *mat = cvCreateMat(src->width, src->height, CV_32FC1);
  for (int i = 0; i < src->width; i++)
  {
    for (int j = 0; j < src->height; j++)
    {
      ve = CV_IMAGE_ELEM(src, uchar, j, i);
      CV_MAT_ELEM(*mat, float, i, j) = (float)ve;
    }
  }
   
  CvMat *rmat = cvCreateMat(src->width, src->height, CV_32FC1);
  CvMat *imat = cvCreateMat(src->width, src->height, CV_32FC1);
   
  switch (Type)
  {
  case CV_GABOR_REAL:
    cvFilter2D( (CvMat*)mat, (CvMat*)mat, (CvMat*)Real, cvPoint( (Width-1)/2, (Width-1)/2));
    break;
  case CV_GABOR_IMAG:
    cvFilter2D( (CvMat*)mat, (CvMat*)mat, (CvMat*)Imag, cvPoint( (Width-1)/2, (Width-1)/2));
    break;
  case CV_GABOR_MAG:
    cvFilter2D( (CvMat*)mat, (CvMat*)rmat, (CvMat*)Real, cvPoint( (Width-1)/2, (Width-1)/2));
    cvFilter2D( (CvMat*)mat, (CvMat*)imat, (CvMat*)Imag, cvPoint( (Width-1)/2, (Width-1)/2));
     
    cvPow(rmat,rmat,2); 
    cvPow(imat,imat,2);
    cvAdd(imat,rmat,mat); 
    cvPow(mat,mat,0.5); 
    break;
  case CV_GABOR_PHASE:
    break;
  }
   
  if (dst->depth == IPL_DEPTH_8U)
  {
    cvNormalize((CvMat*)mat, (CvMat*)mat, 0, 255, CV_MINMAX);
    for (int i = 0; i < mat->rows; i++)
    {
      for (int j = 0; j < mat->cols; j++)
      {
        ve = CV_MAT_ELEM(*mat, float, i, j);
        CV_IMAGE_ELEM(dst, uchar, j, i) = (uchar)cvRound(ve);
      }
    }
  }
   
  if (dst->depth == IPL_DEPTH_32F)
  {
    for (int i = 0; i < mat->rows; i++)
    {
      for (int j = 0; j < mat->cols; j++)
      {
        ve = cvGetReal2D((CvMat*)mat, i, j);
        cvSetReal2D( (IplImage*)dst, j, i, ve );
      }
    }
  }
  cvReleaseMat(&imat);
  cvReleaseMat(&rmat);
  cvReleaseMat(&mat);
}

test.cpp


#include "cv.h"
#include "highgui.h"
#include "Gabor2.h"

int main()
{
CvGabor filter1(0,0);//CvGabor(double dPhi, int iNu)a orientation dPhi, and with a scale iNu




IplImage *kernel = cvCreateImage( cvSize(filter1.get_mask_width(), filter1.get_mask_width()), IPL_DEPTH_8U, 1);
kernel = filter1.get_image(CV_GABOR_REAL);
cvNamedWindow("Gabor Kernel", 1);
cvShowImage("Gabor Kernel", kernel);
cvWaitKey(0);

return 0;

}
  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值