SIFT算法C语言逐步实现详解(上)

 教你一步一步用c语言实现sift算法、上

作者:July、二零一一年三月十二日
出处:http://blog.csdn.net/v_JULY_v
参考:Rob Hess维护的sift 库
环境:windows xp+vc6.0
条件:c语言实现。
说明:本BLOG内会陆续一一实现所有经典算法。
------------------------

引言:
在我写的关于sift算法的前倆篇文章里头,已经对sift算法有了初步的介绍:九、图像特征提取与匹配之SIFT算法,而后在:九(续)、sift算法的编译与实现里,我也简单记录下了如何利用opencv,gsl等库编译运行sift程序。
但据一朋友表示,是否能用c语言实现sift算法,同时,尽量不用到opencv,gsl等第三方库之类的东西。而且,Rob Hess维护的sift 库,也不好懂,有的人根本搞不懂是怎么一回事。
那么本文,就教你如何利用c语言一步一步实现sift算法,同时,你也就能真正明白sift算法到底是怎么一回事了。

ok,先看一下,本程序最终运行的效果图,sift 算法分为五个步骤(下文详述),对应以下第二--第六幅图

sift算法的步骤
要实现一个算法,首先要完全理解这个算法的原理或思想。咱们先来简单了解下,什么叫sift算法:
sift,尺度不变特征转换,是一种电脑视觉的算法用来侦测与描述影像中的局部性特征,它在空间尺度中寻找极值点,并提取出其位置、尺度、旋转不变量,此算法由 David Lowe 在1999年所发表,2004年完善总结。

所谓,Sift算法就是用不同尺度(标准差)的高斯函数对图像进行平滑,然后比较平滑后图像的差别,
差别大的像素就是特征明显的点。

以下是sift算法的五个步骤:
一、建立图像尺度空间(或高斯金字塔),并检测极值点

首先建立尺度空间,要使得图像具有尺度空间不变形,就要建立尺度空间,sift算法采用了高斯函数来建立尺度空间,高斯函数公式为:

上述公式G(x,y,e),即为尺度可变高斯函数。

而,一个图像的尺度空间L(x,y,e) ,定义为原始图像I(x,y)与上述的一个可变尺度的2维高斯函数G(x,y,e) 卷积运算。
即,原始影像I(x,y)在不同的尺度e下,与高斯函数G(x,y,e)进行卷积,得到L(x,y,e),如下:

以上的(x,y)是空间坐标, e,是尺度坐标,或尺度空间因子,e的大小决定平滑程度,大尺度对应图像的概貌特征,小尺度对应图像的细节特征。大的e值对应粗糙尺度(低分辨率),反之,对应精细尺度(高分辨率)。

尺度,受e这个参数控制的表示。而不同的L(x,y,e)就构成了尺度空间,具体计算的时候,即使连续的高斯函数,都被离散为(一般为奇数大小)(2*k+1) *(2*k+1)矩阵,来和数字图像进行卷积运算。
随着e的变化,建立起不同的尺度空间,或称之为建立起图像的高斯金字塔。

像上述L(x,y,e) = G(x,y,e)*I(x,y)的操作,在进行高斯卷积时,整个图像就要遍历所有的像素进行卷积(边界点除外),于此,就造成了时间和空间上的很大浪费。
为了更有效的在尺度空间检测到稳定的关键点,也为了缩小时间和空间复杂度,对上述的操作作了一个改建:即,提出了高斯差分尺度空间(DOG scale-space)。利用不同尺度的高斯差分与原始图像I(x,y)相乘 ,卷积生成。

DOG算子计算简单,是尺度归一化的LOG算子的近似。

ok,耐心点,咱们再来总结一下上述内容:
1、高斯卷积
在组建一组尺度空间后,再组建下一组尺度空间,对上一组尺度空间的最后一幅图像进行二分之一采样,得到下一组尺度空间的第一幅图像,然后进行像建立第一组尺度空间那样的操作,得到第二组尺度空间,公式定义为
L(x,y,e) = G(x,y,e)*I(x,y)

图像金字塔的构建:图像金字塔共O组,每组有S层,下一组的图像由上一组图像降采样得到,效果图,图A如下(左为上一组,右为下一组):


2、高斯差分
在尺度空间建立完毕后,为了能够找到稳定的关键点,采用高斯差分的方法来检测那些在局部位置的极值点,即采用俩个相邻的尺度中的图像相减,即公式定义为:
D(x,y,e) = ((G(x,y,ke) - G(x,y,e)) * I(x,y) 
= L(x,y,ke) - L(x,y,e)

效果图,图B


SIFT的精妙之处在于采用图像金字塔的方法解决这一问题,我们可以把两幅图像想象成是连续的,分别以它们作为底面作四棱锥,就像金字塔,那么每一个 截面与原图像相似,那么两个金字塔中必然会有包含大小一致的物体的无穷个截面,但应用只能是离散的,所以我们只能构造有限层,层数越多当然越好,但处理时 间会相应增加,层数太少不行,因为向下采样的截面中可能找不到尺寸大小一致的两个物体的图像。

咱们再来具体阐述下构造D(x,y,e)的详细步骤:
1、首先采用不同尺度因子的高斯核对图像进行卷积以得到图像的不同尺度空间,将这一组图像作为金子塔图像的第一层。
2、接着对第一层图像中的2倍尺度图像(相对于该层第一幅图像的2倍尺度)以2倍像素距离进行下采样来得到金子塔图像的第二层中的第一幅图像,对该图像采用不同尺度因子的高斯核进行卷积,以获得金字塔图像中第二层的一组图像。
3、再以金字塔图像中第二层中的2倍尺度图像(相对于该层第一幅图像的2倍尺度)以2倍像素距离进行下采样来得到金字塔图像的第三层中的第一幅图像,对该图像采用不同尺度因子的高斯核进行卷积,以获得金字塔图像中第三层的一组图像。这样依次类推,从而获得了金字塔图像的每一层中的一组图像,如下图所示:

4、对上图得到的每一层相邻的高斯图像相减,就得到了高斯差分图像,如下述第一幅图所示。下述第二幅图中的右列显示了将每组中相邻图像相减所生成的高斯差分图像的结果,限于篇幅,图中只给出了第一层和第二层高斯差分图像的计算(下述俩幅图统称为图2):

图像金字塔的建立:对于一幅图像I,建立其在不同尺度(scale)的图像,也成为子八度(octave),这是为了scale-invariant,也就是在任何尺度都能够有对应的特征点,第一个子八度的scale为原图大小,后面每个octave为上一个octave降采样的结果,即原图的1/4(长宽分别减半),构成下一个子八度(高一层金字塔)

5、因为高斯差分函数是归一化的高斯拉普拉斯函数的近似,所以可以从高斯差分金字塔分层结构提取出图像中的极值点作为候选的特征点。对DOG 尺度空间每个点与相邻尺度和相邻位置的点逐个进行比较,得到的局部极值位置即为特征点所处的位置和对应的尺度。

二、检测关键点
为了寻找尺度空间的极值点,每一个采样点要和它所有的相邻点比较,看其是否比它的图像域和尺度域的相邻点大或者小。如下图,图3所示,中间的检测点和它同尺度的8个相邻点和上下相邻尺度对应的9×2个点共26个点比较,以确保在尺度空间和二维图像空间都检测到极值点。

因为需要同相邻尺度进行比较,所以在一组高斯差分图像中只能检测到两个尺度的极值点(如上述第二幅图中右图的五角星标识),而其它尺度的极值点检测则需要在图像金字塔的上一层高斯差分图像中进行。依次类推,最终在图像金字塔中不同层的高斯差分图像中完成不同尺度极值的检测。
当然这样产生的极值点并不都是稳定的特征点,因为某些极值点响应较弱,而且DOG算子会产生较强的边缘响应。

三、关键点方向的分配
为了使描述符具有旋转不变性,需要利用图像的局部特征为给每一个关键点分配一个方向。利用关键点邻域像素的梯度及方向分布的特性,可以得到梯度模值和方向如下:


其中,尺度为每个关键点各自所在的尺度。
在以关键点为中心的邻域窗口内采样,并用直方图统计邻域像素的梯度方向。梯度直方图的范围是0~360度,其中每10度一个方向,总共36个方向。
直方图的峰值则代表了该关键点处邻域梯度的主方向,即作为该关键点的方向。

在计算方向直方图时,需要用一个参数等于关键点所在尺度1.5倍的高斯权重窗对方向直方图进行加权,上图中用蓝色的圆形表示,中心处的蓝色较重,表示权值最大,边缘处颜色潜,表示权值小。如下图所示,该示例中为了简化给出了8方向的方向直方图计算结果,实际sift创始人David Lowe的原论文中采用36方向的直方图。

方向直方图的峰值则代表了该特征点处邻域梯度的方向,以直方图中最大值作为该关键点的主方向。为了增强匹配的鲁棒性,只保留峰值大于主方向峰值80%的方向作为该关键点的辅方向。因此,对于同一梯度值的多个峰值的关键点位置,在相同位置和尺度将会有多个关键点被创建但方向不同。仅有15%的关键点被赋予多个方向,但可以明显的提高关键点匹配的稳定性。

至此,图像的关键点已检测完毕,每个关键点有三个信息:位置、所处尺度、方向。由此可以确定一个SIFT特征区域。

四、特征点描述符
通过以上步骤,对于每一个关键点,拥有三个信息:位置、尺度以及方向。接下来就是为每个关键点建立一个描述符,使其不随各种变化而改变,比如光照变化、视角变化等等。并且描述符应该有较高的独特性,以便于提高特征点正确匹配的概率。 
首先将坐标轴旋转为关键点的方向,以确保旋转不变性。

接下来以关键点为中心取8×8的窗口。
上图,图5中左部分的中央黑点为当前关键点的位置,每个小格代表关键点邻域所在尺度空间的一个像素,箭头方向代表该像素的梯度方向,箭头长度代表梯度模值,图中蓝色的圈代表高斯加权的范围(越靠近关键点的像素梯度方向信息贡献越大)。
然后在每4×4的小块上计算8个方向的梯度方向直方图,绘制每个梯度方向的累加值,即可形成一个种子点,如图5右部分所示。此图中一个关键点由2×2共4个种子点组成,每个种子点有8个方向向量信息。这种邻域方向性信息联合的思想增强了算法抗噪声的能力,同时对于含有定位误差的特征匹配也提供了较好的容错性。

实际计算过程中,为了增强匹配的稳健性,Lowe建议对每个关键点使用4×4共16个种子点来描述,这样对于一个关键点就可以产生128个数据,即最终形成128维的SIFT特征向量。此时SIFT特征向量已经去除了尺度变化、旋转等几何变形因素的影响,再继续将特征向量的长度归一化,则可以进一步去除光照变化的影响。


五、最后一步:当两幅图像的SIFT特征向量生成后,下一步我们采用关键点特征向量的欧式距离来作为两幅图像中关键点的相似性判定度量。取上图中,图像A中的某个关键点,并找出其与图像B中欧式距离最近的前两个关键点,在这两个关键点中,如果最近的距离除以次近的距离少于某个比例阈值,则接受这一对匹配点。降低这个比例阈值,SIFT匹配点数目会减少,但更加稳定。关于sift 算法的更多理论介绍请参看此文:http://blog.csdn.net/abcjennifer/article/details/7639681

sift算法的逐步c实现
ok,上文搅了那么多的理论,如果你没有看懂它,咋办列?没关系,下面,咱们来一步一步实现此sift算法,即使你没有看到上述的理论,慢慢的,你也会明白sift算法到底是怎么一回事,sift算法到底是怎么实现的...。
yeah,请看:

前期工作:
在具体编写核心函数之前,得先做几个前期的准备工作:

0、头文件:


  1. #ifdef _CH_
  2. #pragma package
  3. #endif
  4. #ifndef _EiC
  5. #include
  6. #include "stdlib.h"
  7. #include "string.h"
  8. #include "malloc.h"
  9. #include "math.h"
  10. #include
  11. #include
  12. #include
  13. #include
  14. #include
  15. #include
  16. #include
  17. #endif
  18. #ifdef _EiC
  19. #define WIN32
  20. #endif
[cpp]  view plain  copy
  1. #ifdef _CH_  
  2. #pragma package   
  3. #endif  
  4.   
  5. #ifndef _EiC  
  6. #include   
  7.   
  8. #include "stdlib.h"  
  9. #include "string.h"   
  10. #include "malloc.h"   
  11. #include "math.h"   
  12. #include   
  13. #include   
  14. #include   
  15. #include   
  16. #include   
  17. #include   
  18. #include   
  19. #endif  
  20.   
  21. #ifdef _EiC  
  22. #define WIN32  
  23. #endif  

1、定义几个宏,及变量,以免下文函数中,突然冒出一个变量,而您却不知道怎么一回事:

[cpp]  view plain copy
  1. #define NUMSIZE 2
  2. #define GAUSSKERN 3.5
  3. #define PI 3.14159265358979323846
  4. //Sigma of base image -- See D.L.'s paper.
  5. #define INITSIGMA 0.5
  6. //Sigma of each octave -- See D.L.'s paper.
  7. #define SIGMA sqrt(3)//1.6//
  8. //Number of scales per octave. See D.L.'s paper.
  9. #define SCALESPEROCTAVE 2
  10. #define MAXOCTAVES 4
  11. int numoctaves;
  12. #define CONTRAST_THRESHOLD 0.02
  13. #define CURVATURE_THRESHOLD 10.0
  14. #define DOUBLE_BASE_IMAGE_SIZE 1
  15. #define peakRelThresh 0.8
  16. #define LEN 128
  17. // temporary storage
  18. CvMemStorage* storage = 0;
[cpp]  view plain  copy
  1. #define NUMSIZE 2  
  2. #define GAUSSKERN 3.5  
  3. #define PI 3.14159265358979323846  
  4.   
  5. //Sigma of base image -- See D.L.'s paper.  
  6. #define INITSIGMA 0.5  
  7. //Sigma of each octave -- See D.L.'s paper.  
  8. #define SIGMA sqrt(3)//1.6//  
  9.   
  10. //Number of scales per octave.  See D.L.'s paper.  
  11. #define SCALESPEROCTAVE 2  
  12. #define MAXOCTAVES 4  
  13. int     numoctaves;  
  14.   
  15. #define CONTRAST_THRESHOLD   0.02  
  16. #define CURVATURE_THRESHOLD  10.0  
  17. #define DOUBLE_BASE_IMAGE_SIZE 1  
  18. #define peakRelThresh 0.8  
  19. #define LEN 128  
  20.   
  21. // temporary storage  
  22. CvMemStorage* storage = 0;   

2、然后,咱们还得,声明几个变量,以及建几个数据结构(数据结构是一切程序事物的基础麻。):

  1. //Data structure for a float image.
  2. typedef struct ImageSt { /*金字塔每一层*/
  3. float levelsigma;
  4. int levelsigmalength;
  5. float absolute_sigma;
  6. CvMat *Level; //CvMat是OPENCV的矩阵类,其元素可以是图像的象素值
  7. } ImageLevels;
  8. typedef struct ImageSt1 { /*金字塔每一阶梯*/
  9. int row, col; //Dimensions of image.
  10. float subsample;
  11. ImageLevels *Octave;
  12. } ImageOctaves;
  13. ImageOctaves *DOGoctaves;
  14. //DOG pyr,DOG算子计算简单,是尺度归一化的LoG算子的近似。
  15. ImageOctaves *mag_thresh ;
  16. ImageOctaves *mag_pyr ;
  17. ImageOctaves *grad_pyr ;
  18. //keypoint数据结构,Lists of keypoints are linked by the "next" field.
  19. typedef struct KeypointSt
  20. {
  21. float row, col; /* 反馈回原图像大小,特征点的位置 */
  22. float sx,sy; /* 金字塔中特征点的位置*/
  23. int octave,level;/*金字塔中,特征点所在的阶梯、层次*/
  24. float scale, ori,mag; /*所在层的尺度sigma,主方向orientation (range [-PI,PI]),以及幅值*/
  25. float *descrip; /*特征描述字指针:128维或32维等*/
  26. struct KeypointSt *next;/* Pointer to next keypoint in list. */
  27. } *Keypoint;
  28. //定义特征点具体变量
  29. Keypoint keypoints=NULL; //用于临时存储特征点的位置等
  30. Keypoint keyDescriptors=NULL; //用于最后的确定特征点以及特征描述字
[cpp]  view plain  copy
  1. //Data structure for a float image.  
  2. typedef struct ImageSt {        /*金字塔每一层*/  
  3.    
  4.  float levelsigma;  
  5.  int levelsigmalength;  
  6.  float absolute_sigma;  
  7.  CvMat *Level;       //CvMat是OPENCV的矩阵类,其元素可以是图像的象素值         
  8. } ImageLevels;  
  9.   
  10. typedef struct ImageSt1 {      /*金字塔每一阶梯*/  
  11.  int row, col;          //Dimensions of image.   
  12.  float subsample;  
  13.  ImageLevels *Octave;                
  14. } ImageOctaves;  
  15.   
  16. ImageOctaves *DOGoctaves;        
  17. //DOG pyr,DOG算子计算简单,是尺度归一化的LoG算子的近似。  
  18.     
  19. ImageOctaves *mag_thresh ;  
  20. ImageOctaves *mag_pyr ;  
  21. ImageOctaves *grad_pyr ;  
  22.   
  23. //keypoint数据结构,Lists of keypoints are linked by the "next" field.  
  24. typedef struct KeypointSt   
  25. {  
  26.  float row, col; /* 反馈回原图像大小,特征点的位置 */  
  27.  float sx,sy;    /* 金字塔中特征点的位置*/  
  28.  int octave,level;/*金字塔中,特征点所在的阶梯、层次*/  
  29.    
  30.  float scale, ori,mag; /*所在层的尺度sigma,主方向orientation (range [-PI,PI]),以及幅值*/  
  31.  float *descrip;       /*特征描述字指针:128维或32维等*/  
  32.  struct KeypointSt *next;/* Pointer to next keypoint in list. */  
  33. } *Keypoint;  
  34.   
  35. //定义特征点具体变量  
  36. Keypoint keypoints=NULL;      //用于临时存储特征点的位置等  
  37. Keypoint keyDescriptors=NULL; //用于最后的确定特征点以及特征描述字  

3、声明几个图像的基本处理函数:


  1. CvMat * halfSizeImage(CvMat * im); //缩小图像:下采样
  2. CvMat * doubleSizeImage(CvMat * im); //扩大图像:最近临方法
  3. CvMat * doubleSizeImage2(CvMat * im); //扩大图像:线性插值
  4. float getPixelBI(CvMat * im, float col, float row);//双线性插值函数
  5. void normalizeVec(float* vec, int dim);//向量归一化
  6. CvMat* GaussianKernel2D(float sigma); //得到2维高斯核
  7. void normalizeMat(CvMat* mat) ; //矩阵归一化
  8. float* GaussianKernel1D(float sigma, int dim) ; //得到1维高斯核
  9. //在具体像素处宽度方向进行高斯卷积
  10. float ConvolveLocWidth(float* kernel, int dim, CvMat * src, int x, int y) ;
  11. //在整个图像宽度方向进行1D高斯卷积
  12. void Convolve1DWidth(float* kern, int dim, CvMat * src, CvMat * dst) ;
  13. //在具体像素处高度方向进行高斯卷积
  14. float ConvolveLocHeight(float* kernel, int dim, CvMat * src, int x, int y) ;
  15. //在整个图像高度方向进行1D高斯卷积
  16. void Convolve1DHeight(float* kern, int dim, CvMat * src, CvMat * dst);
  17. //用高斯函数模糊图像
  18. int BlurImage(CvMat * src, CvMat * dst, float sigma) ;
[cpp]  view plain  copy
  1. CvMat * halfSizeImage(CvMat * im);     //缩小图像:下采样  
  2. CvMat * doubleSizeImage(CvMat * im);   //扩大图像:最近临方法  
  3. CvMat * doubleSizeImage2(CvMat * im);  //扩大图像:线性插值  
  4. float getPixelBI(CvMat * im, float col, float row);//双线性插值函数  
  5. void normalizeVec(float* vec, int dim);//向量归一化    
  6. CvMat* GaussianKernel2D(float sigma);  //得到2维高斯核  
  7. void normalizeMat(CvMat* mat) ;        //矩阵归一化  
  8. float* GaussianKernel1D(float sigma, int dim) ; //得到1维高斯核  
  9.   
  10. //在具体像素处宽度方向进行高斯卷积  
  11. float ConvolveLocWidth(float* kernel, int dim, CvMat * src, int x, int y) ;    
  12. //在整个图像宽度方向进行1D高斯卷积  
  13. void Convolve1DWidth(float* kern, int dim, CvMat * src, CvMat * dst) ;         
  14. //在具体像素处高度方向进行高斯卷积  
  15. float ConvolveLocHeight(float* kernel, int dim, CvMat * src, int x, int y) ;   
  16. //在整个图像高度方向进行1D高斯卷积  
  17. void Convolve1DHeight(float* kern, int dim, CvMat * src, CvMat * dst);       
  18. //用高斯函数模糊图像    
  19. int BlurImage(CvMat * src, CvMat * dst, float sigma) ;            

算法核心
本程序中,sift算法被分为以下五个步骤及其相对应的函数(可能表述与上,或与前俩篇文章有所偏差,但都一个意思):
[cpp]  view plain copy
  1. //SIFT算法第一步:图像预处理
  2. CvMat *ScaleInitImage(CvMat * im) ; //金字塔初始化
  3. //SIFT算法第二步:建立高斯金字塔函数
  4. ImageOctaves* BuildGaussianOctaves(CvMat * image) ; //建立高斯金字塔
  5. //SIFT算法第三步:特征点位置检测,最后确定特征点的位置
  6. int DetectKeypoint(int numoctaves, ImageOctaves *GaussianPyr);
  7. void DisplayKeypointLocation(IplImage* image, ImageOctaves *GaussianPyr);
  8. //SIFT算法第四步:计算高斯图像的梯度方向和幅值,计算各个特征点的主方向
  9. void ComputeGrad_DirecandMag(int numoctaves, ImageOctaves *GaussianPyr);
  10. int FindClosestRotationBin (int binCount, float angle); //进行方向直方图统计
  11. void AverageWeakBins (double* bins, int binCount); //对方向直方图滤波
  12. //确定真正的主方向
  13. bool InterpolateOrientation (double left, double middle,double right, double*degreeCorrection, double *peakValue);
  14. //确定各个特征点处的主方向函数
  15. void AssignTheMainOrientation(int numoctaves, ImageOctaves *GaussianPyr,ImageOctaves *mag_pyr,ImageOctaves *grad_pyr);
  16. //显示主方向
  17. void DisplayOrientation (IplImage* image, ImageOctaves *GaussianPyr);
  18. //SIFT算法第五步:抽取各个特征点处的特征描述字
  19. void ExtractFeatureDescriptors(int numoctaves, ImageOctaves *GaussianPyr);
  20. //为了显示图象金字塔,而作的图像水平、垂直拼接
  21. CvMat* MosaicHorizen( CvMat* im1, CvMat* im2 );
  22. CvMat* MosaicVertical( CvMat* im1, CvMat* im2 );
  23. //特征描述点,网格
  24. #define GridSpacing 4
[cpp]  view plain  copy
  1. //SIFT算法第一步:图像预处理  
  2. CvMat *ScaleInitImage(CvMat * im) ;                  //金字塔初始化  
  3.   
  4. //SIFT算法第二步:建立高斯金字塔函数  
  5. ImageOctaves* BuildGaussianOctaves(CvMat * image) ;  //建立高斯金字塔  
  6.   
  7. //SIFT算法第三步:特征点位置检测,最后确定特征点的位置  
  8. int DetectKeypoint(int numoctaves, ImageOctaves *GaussianPyr);  
  9. void DisplayKeypointLocation(IplImage* image, ImageOctaves *GaussianPyr);  
  10.   
  11. //SIFT算法第四步:计算高斯图像的梯度方向和幅值,计算各个特征点的主方向  
  12. void ComputeGrad_DirecandMag(int numoctaves, ImageOctaves *GaussianPyr);  
  13.   
  14. int FindClosestRotationBin (int binCount, float angle);  //进行方向直方图统计  
  15. void AverageWeakBins (double* bins, int binCount);       //对方向直方图滤波  
  16. //确定真正的主方向  
  17. bool InterpolateOrientation (double left, double middle,double right, double *degreeCorrection, double *peakValue);  
  18. //确定各个特征点处的主方向函数  
  19. void AssignTheMainOrientation(int numoctaves, ImageOctaves *GaussianPyr,ImageOctaves *mag_pyr,ImageOctaves *grad_pyr);  
  20. //显示主方向  
  21. void DisplayOrientation (IplImage* image, ImageOctaves *GaussianPyr);  
  22.   
  23. //SIFT算法第五步:抽取各个特征点处的特征描述字  
  24. void ExtractFeatureDescriptors(int numoctaves, ImageOctaves *GaussianPyr);  
  25.   
  26. //为了显示图象金字塔,而作的图像水平、垂直拼接  
  27. CvMat* MosaicHorizen( CvMat* im1, CvMat* im2 );  
  28. CvMat* MosaicVertical( CvMat* im1, CvMat* im2 );  
  29.   
  30. //特征描述点,网格    
  31. #define GridSpacing 4  


主体实现
ok,以上所有的工作都就绪以后,那么接下来,咱们就先来编写main函数,因为你一看主函数之后,你就立马能发现sift算法的工作流程及其原理了。
(主函数中涉及到的函数,下一篇文章:一、教你一步一步用c语言实现sift算法、下,咱们自会一个一个编写):


  1. int main( void )
  2. {
  3. //声明当前帧IplImage指针
  4. IplImage* src = NULL;
  5. IplImage* image1 = NULL;
  6. IplImage* grey_im1 = NULL;
  7. IplImage* DoubleSizeImage = NULL;
  8. IplImage* mosaic1 = NULL;
  9. IplImage* mosaic2 = NULL;
  10. CvMat* mosaicHorizen1 = NULL;
  11. CvMat* mosaicHorizen2 = NULL;
  12. CvMat* mosaicVertical1 = NULL;
  13. CvMat* image1Mat = NULL;
  14. CvMat* tempMat=NULL;
  15. ImageOctaves *Gaussianpyr;
  16. int rows,cols;
  17. #define Im1Mat(ROW,COL) ((float *)(image1Mat->data.fl + image1Mat->step/sizeof(float) *(ROW)))[(COL)]
  18. //灰度图象像素的数据结构
  19. #define Im1B(ROW,COL) ((uchar*)(image1->imageData + image1->widthStep*(ROW)))[(COL)*3]
  20. #define Im1G(ROW,COL) ((uchar*)(image1->imageData + image1->widthStep*(ROW)))[(COL)*3+1]
  21. #define Im1R(ROW,COL) ((uchar*)(image1->imageData + image1->widthStep*(ROW)))[(COL)*3+2]
  22. storage = cvCreateMemStorage(0);
  23. //读取图片
  24. if( (src = cvLoadImage( "street1.jpg", 1)) == 0 ) // test1.jpg einstein.pgm back1.bmp
  25. return -1;
  26. //为图像分配内存
  27. image1 = cvCreateImage(cvSize(src->width, src->height), IPL_DEPTH_8U,3);
  28. grey_im1 = cvCreateImage(cvSize(src->width, src->height), IPL_DEPTH_8U,1);
  29. DoubleSizeImage = cvCreateImage(cvSize(2*(src->width), 2*(src->height)), IPL_DEPTH_8U,3);
  30. //为图像阵列分配内存,假设两幅图像的大小相同,tempMat跟随image1的大小
  31. image1Mat = cvCreateMat(src->height, src->width, CV_32FC1);
  32. //转化成单通道图像再处理
  33. cvCvtColor(src, grey_im1, CV_BGR2GRAY);
  34. //转换进入Mat数据结构,图像操作使用的是浮点型操作
  35. cvConvert(grey_im1, image1Mat);
  36. double t = (double)cvGetTickCount();
  37. //图像归一化
  38. cvConvertScale( image1Mat, image1Mat, 1.0/255, 0 );
  39. int dim = min(image1Mat->rows, image1Mat->cols);
  40. numoctaves = (int) (log((double) dim) / log(2.0)) - 2; //金字塔阶数
  41. numoctaves = min(numoctaves, MAXOCTAVES);
  42. //SIFT算法第一步,预滤波除噪声,建立金字塔底层
  43. tempMat = ScaleInitImage(image1Mat) ;
  44. //SIFT算法第二步,建立Guassian金字塔和DOG金字塔
  45. Gaussianpyr = BuildGaussianOctaves(tempMat) ;
  46. t = (double)cvGetTickCount() - t;
  47. printf( "the time of build Gaussian pyramid and DOG pyramid is %.1f/n", t/(cvGetTickFrequency()*1000.) );
  48. #define ImLevels(OCTAVE,LEVEL,ROW,COL) ((float *)(Gaussianpyr[(OCTAVE)].Octave[(LEVEL)].Level->data.fl + Gaussianpyr[(OCTAVE)].Octave[(LEVEL)].Level->step/sizeof(float) *(ROW)))[(COL)]
  49. //显示高斯金字塔
  50. for (int i=0; i
  51. {
  52. if (i==0)
  53. {
  54. mosaicHorizen1=MosaicHorizen( (Gaussianpyr[0].Octave)[0].Level, (Gaussianpyr[0].Octave)[1].Level );
  55. for (int j=2;j
  56. mosaicHorizen1=MosaicHorizen( mosaicHorizen1, (Gaussianpyr[0].Octave)[j].Level );
  57. for ( j=0;j
  58. mosaicHorizen1=halfSizeImage(mosaicHorizen1);
  59. }
  60. else if (i==1)
  61. {
  62. mosaicHorizen2=MosaicHorizen( (Gaussianpyr[1].Octave)[0].Level, (Gaussianpyr[1].Octave)[1].Level );
  63. for (int j=2;j
  64. mosaicHorizen2=MosaicHorizen( mosaicHorizen2, (Gaussianpyr[1].Octave)[j].Level );
  65. for ( j=0;j
  66. mosaicHorizen2=halfSizeImage(mosaicHorizen2);
  67. mosaicVertical1=MosaicVertical( mosaicHorizen1, mosaicHorizen2 );
  68. }
  69. else
  70. {
  71. mosaicHorizen1=MosaicHorizen( (Gaussianpyr[i].Octave)[0].Level, (Gaussianpyr[i].Octave)[1].Level );
  72. for (int j=2;j
  73. mosaicHorizen1=MosaicHorizen( mosaicHorizen1, (Gaussianpyr[i].Octave)[j].Level );
  74. for ( j=0;j
  75. mosaicHorizen1=halfSizeImage(mosaicHorizen1);
  76. mosaicVertical1=MosaicVertical( mosaicVertical1, mosaicHorizen1 );
  77. }
  78. }
  79. mosaic1 = cvCreateImage(cvSize(mosaicVertical1->width, mosaicVertical1->height), IPL_DEPTH_8U,1);
  80. cvConvertScale( mosaicVertical1, mosaicVertical1, 255.0, 0 );
  81. cvConvertScaleAbs( mosaicVertical1, mosaic1, 1, 0 );
  82. // cvSaveImage("GaussianPyramid of me.jpg",mosaic1);
  83. cvNamedWindow("mosaic1",1);
  84. cvShowImage("mosaic1", mosaic1);
  85. cvWaitKey(0);
  86. cvDestroyWindow("mosaic1");
  87. //显示DOG金字塔
  88. for ( i=0; i
  89. {
  90. if (i==0)
  91. {
  92. mosaicHorizen1=MosaicHorizen( (DOGoctaves[0].Octave)[0].Level, (DOGoctaves[0].Octave)[1].Level );
  93. for (int j=2;j
  94. mosaicHorizen1=MosaicHorizen( mosaicHorizen1, (DOGoctaves[0].Octave)[j].Level );
  95. for ( j=0;j
  96. mosaicHorizen1=halfSizeImage(mosaicHorizen1);
  97. }
  98. else if (i==1)
  99. {
  100. mosaicHorizen2=MosaicHorizen( (DOGoctaves[1].Octave)[0].Level, (DOGoctaves[1].Octave)[1].Level );
  101. for (int j=2;j
  102. mosaicHorizen2=MosaicHorizen( mosaicHorizen2, (DOGoctaves[1].Octave)[j].Level );
  103. for ( j=0;j
  104. mosaicHorizen2=halfSizeImage(mosaicHorizen2);
  105. mosaicVertical1=MosaicVertical( mosaicHorizen1, mosaicHorizen2 );
  106. }
  107. else
  108. {
  109. mosaicHorizen1=MosaicHorizen( (DOGoctaves[i].Octave)[0].Level, (DOGoctaves[i].Octave)[1].Level );
  110. for (int j=2;j
  111. mosaicHorizen1=MosaicHorizen( mosaicHorizen1, (DOGoctaves[i].Octave)[j].Level );
  112. for ( j=0;j
  113. mosaicHorizen1=halfSizeImage(mosaicHorizen1);
  114. mosaicVertical1=MosaicVertical( mosaicVertical1, mosaicHorizen1 );
  115. }
  116. }
  117. //考虑到DOG金字塔各层图像都会有正负,所以,必须寻找最负的,以将所有图像抬高一个台阶去显示
  118. double min_val=0;
  119. double max_val=0;
  120. cvMinMaxLoc( mosaicVertical1, &min_val, &max_val,NULL, NULL, NULL );
  121. if ( min_val<0.0 )
  122. cvAddS( mosaicVertical1, cvScalarAll( (-1.0)*min_val ), mosaicVertical1, NULL );
  123. mosaic2 = cvCreateImage(cvSize(mosaicVertical1->width, mosaicVertical1->height), IPL_DEPTH_8U,1);
  124. cvConvertScale( mosaicVertical1, mosaicVertical1, 255.0/(max_val-min_val), 0 );
  125. cvConvertScaleAbs( mosaicVertical1, mosaic2, 1, 0 );
  126. // cvSaveImage("DOGPyramid of me.jpg",mosaic2);
  127. cvNamedWindow("mosaic1",1);
  128. cvShowImage("mosaic1", mosaic2);
  129. cvWaitKey(0);
  130. //SIFT算法第三步:特征点位置检测,最后确定特征点的位置
  131. int keycount=DetectKeypoint(numoctaves, Gaussianpyr);
  132. printf("the keypoints number are %d ;/n", keycount);
  133. cvCopy(src,image1,NULL);
  134. DisplayKeypointLocation( image1 ,Gaussianpyr);
  135. cvPyrUp( image1, DoubleSizeImage, CV_GAUSSIAN_5x5 );
  136. cvNamedWindow("image1",1);
  137. cvShowImage("image1", DoubleSizeImage);
  138. cvWaitKey(0);
  139. cvDestroyWindow("image1");
  140. //SIFT算法第四步:计算高斯图像的梯度方向和幅值,计算各个特征点的主方向
  141. ComputeGrad_DirecandMag(numoctaves, Gaussianpyr);
  142. AssignTheMainOrientation( numoctaves, Gaussianpyr,mag_pyr,grad_pyr);
  143. cvCopy(src,image1,NULL);
  144. DisplayOrientation ( image1, Gaussianpyr);
  145. // cvPyrUp( image1, DoubleSizeImage, CV_GAUSSIAN_5x5 );
  146. cvNamedWindow("image1",1);
  147. // cvResizeWindow("image1", 2*(image1->width), 2*(image1->height) );
  148. cvShowImage("image1", image1);
  149. cvWaitKey(0);
  150. //SIFT算法第五步:抽取各个特征点处的特征描述字
  151. ExtractFeatureDescriptors( numoctaves, Gaussianpyr);
  152. cvWaitKey(0);
  153. //销毁窗口
  154. cvDestroyWindow("image1");
  155. cvDestroyWindow("mosaic1");
  156. //释放图像
  157. cvReleaseImage(&image1);
  158. cvReleaseImage(&grey_im1);
  159. cvReleaseImage(&mosaic1);
  160. cvReleaseImage(&mosaic2);
  161. return 0;
  162. }

你可能感兴趣的:(其他)