SURF源码分析之main.cpp

main.cpp主要由几个主函数组成，通过选择PROCEDURE的值可进行下列6个操作：

1、特征点检测
2、视频特征点检测
3、图像与视频匹配
4、动点查询
5、图像与图像匹配
6、特征点聚类-Kmeas算法

/*********************************************************** *  --- OpenSURF ---                                       **  This library is distributed under the GNU GPL. Please   **  use the contact form at http://www.chrisevansdev.com    **  for more information.                                   **                                                          **  C. Evans, Research Into Robust Visual Features,         **  MSc University of Bristol, 2008.                        **                                                          *************************************************************/#include "surflib.h"#include "kmeans.h"#include <ctime>#include <iostream>//-------------------------------------------------------// In order to you use OpenSURF, the following illustrates// some of the simple tasks you can do.  It takes only 1// function call to extract described SURF features!// Define PROCEDURE as://  - 1 and supply image path to run on static image//  - 2 to capture from a webcam//  - 3 to match find an object in an image (work in progress)//  - 4 to display moving features (work in progress)//  - 5 to show matches between static images#define PROCEDURE 2//-------------------------------------------------------// 视频特征点检测int mainImage(void){  // Declare Ipoints and other stuff  IpVec ipts;//特征点容器  IplImage *img=cvLoadImage("imgs/sf.jpg");//待提取特征点图像  // Detect and describe interest points in the image  clock_t start = clock();//特征点提取计时开始  surfDetDes(img, ipts, false, 5, 4, 2, 0.0004f);//特征点提取函数   clock_t end = clock();//特征点提取计时结束  std::cout<< "OpenSURF found: " << ipts.size() << " interest points" << std::endl;  std::cout<< "OpenSURF took: " << float(end - start) / CLOCKS_PER_SEC  << " seconds" << std::endl;  // Draw the detected points  drawIpoints(img, ipts);//在图像中画出特征点    // Display the result  showImage(img);//显示图像  return 0;}//-------------------------------------------------------// 视频图像特征点检测int mainVideo(void){  // Initialise capture device  CvCapture* capture = cvCreateFileCapture( "E:\\photo\\毕业照\\杀人.mov " );  //CvCapture* capture = cvCaptureFromCAM( CV_CAP_ANY );//从摄像头获取视频  if(!capture) error("No Capture");//检测是否获取视频  // Initialise video writer  //cv::VideoWriter vw("c:\\out.avi", CV_FOURCC('D','I','V','X'),10,cvSize(320,240),1);  //vw << img;  // Create a window   cvNamedWindow("OpenSURF", CV_WINDOW_AUTOSIZE );//创建显示窗口  // Declare Ipoints and other stuff  IpVec ipts;//特征点容器  IplImage *img=NULL;//获取帧图像  // Main capture loop  while( 1 )   {    // Grab frame from the capture source    img = cvQueryFrame(capture);//将帧图像赋值给img    // Extract surf points    surfDetDes(img, ipts, false, 4, 4, 2, 0.004f);//获取图像SURF特征值        // Draw the detected points    drawIpoints(img, ipts);//在图像中画出特征点    // Draw the FPS figure    drawFPS(img);// 显示帧数    // Display the result    cvShowImage("OpenSURF", img);//显示图片    // If ESC key pressed exit loop    if( (cvWaitKey(10) & 255) == 27 ) break;  }  cvReleaseCapture( &capture );//释放视频  cvDestroyWindow( "OpenSURF" );//摧毁窗口  return 0;}//-------------------------------------------------------int mainMatch(void){  // Initialise capture device  CvCapture* capture = cvCaptureFromCAM( CV_CAP_ANY );//从摄像头获取图像  if(!capture) error("No Capture");//检测是否成功获取视频  // Declare Ipoints and other stuff  IpPairVec matches;//匹配特征点容器  IpVec ipts, ref_ipts;//特征点容器    // This is the reference object we wish to find in video frame  // Replace the line below with IplImage *img = cvLoadImage("imgs/object.jpg");   // where object.jpg is the planar object to be located in the video  IplImage *img = cvLoadImage("imgs/object.jpg"); //待检测的图像  if (img == NULL) error("Need to load reference image in order to run matching procedure");  CvPoint src_corners[4] = {{0,0}, {img->width,0}, {img->width, img->height}, {0, img->height}};  CvPoint dst_corners[4];  // Extract reference object Ipoints  surfDetDes(img, ref_ipts, false, 3, 4, 3, 0.004f);  drawIpoints(img, ref_ipts);  showImage(img);  // Create a window   cvNamedWindow("OpenSURF", CV_WINDOW_AUTOSIZE );  // Main capture loop  while( true )   {    // Grab frame from the capture source    img = cvQueryFrame(capture);         // Detect and describe interest points in the frame    surfDetDes(img, ipts, false, 3, 4, 3, 0.004f);    // Fill match vector    getMatches(ipts,ref_ipts,matches);        // This call finds where the object corners should be in the frame    if (translateCorners(matches, src_corners, dst_corners))    {      // Draw box around object      for(int i = 0; i < 4; i++ )      {        CvPoint r1 = dst_corners[i%4];        CvPoint r2 = dst_corners[(i+1)%4];        cvLine( img, cvPoint(r1.x, r1.y),          cvPoint(r2.x, r2.y), cvScalar(255,255,255), 3 );      }      for (unsigned int i = 0; i < matches.size(); ++i)        drawIpoint(img, matches[i].first);    }    // Draw the FPS figure    drawFPS(img);    // Display the result    cvShowImage("OpenSURF", img);    // If ESC key pressed exit loop    if( (cvWaitKey(10) & 255) == 27 ) break;  }  // Release the capture device  cvReleaseCapture( &capture );  cvDestroyWindow( "OpenSURF" );  return 0;}//-------------------------------------------------------//动点检测int mainMotionPoints(void){  // Initialise capture device  CvCapture* capture = cvCaptureFromCAM( CV_CAP_ANY );  if(!capture) error("No Capture");  // Create a window   cvNamedWindow("OpenSURF", CV_WINDOW_AUTOSIZE );  // Declare Ipoints and other stuff  IpVec ipts, old_ipts, motion;  IpPairVec matches;  IplImage *img;  // Main capture loop  while( 1 )   {    // Grab frame from the capture source    img = cvQueryFrame(capture);    // Detect and describe interest points in the image    old_ipts = ipts;    surfDetDes(img, ipts, true, 3, 4, 2, 0.0004f);    // Fill match vector    getMatches(ipts,old_ipts,matches);    for (unsigned int i = 0; i < matches.size(); ++i)     {      const float & dx = matches[i].first.dx;      const float & dy = matches[i].first.dy;      float speed = sqrt(dx*dx+dy*dy);      if (speed > 5 && speed < 30)         drawIpoint(img, matches[i].first, 3);    }            // Display the result    cvShowImage("OpenSURF", img);    // If ESC key pressed exit loop    if( (cvWaitKey(10) & 255) == 27 ) break;  }  // Release the capture device  cvReleaseCapture( &capture );  cvDestroyWindow( "OpenSURF" );  return 0;}//-------------------------------------------------------int mainStaticMatch(){  IplImage *img1, *img2;  clock_t start = clock();  //img1 = cvLoadImage("data/landscape-a.jpg");  //img2 = cvLoadImage("data/landscape-b.jpg");  img1 = cvLoadImage("E:\\photo\\测试\\test1.jpg");  img2 = cvLoadImage("E:\\photo\\测试\\test2.jpg");  IpVec ipts1, ipts2;  surfDetDes(img1,ipts1,false,4,4,2,0.0001f);  surfDetDes(img2,ipts2,false,4,4,2,0.0001f);  IpPairVec matches;  getMatches(ipts1,ipts2,matches);  for (unsigned int i = 0; i < matches.size(); ++i)  {    drawPoint(img1,matches[i].first);    drawPoint(img2,matches[i].second);      const int & w = img1->width;    cvLine(img1,cvPoint(matches[i].first.x,matches[i].first.y),cvPoint(matches[i].second.x+w,matches[i].second.y), cvScalar(255,255,255),1);    cvLine(img2,cvPoint(matches[i].first.x-w,matches[i].first.y),cvPoint(matches[i].second.x,matches[i].second.y), cvScalar(255,255,255),1);  }  std::cout<< "Matches: " << matches.size();  clock_t end = clock();  std::cout<< "OpenSURF took: " << float(end - start) / CLOCKS_PER_SEC  << " seconds" << std::endl;  cvNamedWindow("1", CV_WINDOW_AUTOSIZE );  cvNamedWindow("2", CV_WINDOW_AUTOSIZE );  cvShowImage("1", img1);  cvShowImage("2",img2);  cvWaitKey(0);  return 0;}//-------------------------------------------------------// 特征点聚类-Kmeas算法int mainKmeans(void){  IplImage *img = cvLoadImage("imgs/img1.jpg");  IpVec ipts;  Kmeans km;    // Get Ipoints  surfDetDes(img,ipts,true,3,4,2,0.0006f);  for (int repeat = 0; repeat < 10; ++repeat)  {    IplImage *img = cvLoadImage("imgs/img1.jpg");    km.Run(&ipts, 5, true);    drawPoints(img, km.clusters);    for (unsigned int i = 0; i < ipts.size(); ++i)    {      cvLine(img, cvPoint(ipts[i].x,ipts[i].y), cvPoint(km.clusters[ipts[i].clusterIndex].x ,km.clusters[ipts[i].clusterIndex].y),cvScalar(255,255,255));    }    showImage(img);  }  return 0;}//-------------------------------------------------------int main(void) {  if (PROCEDURE == 1) return mainImage();//特征点检测  if (PROCEDURE == 2) return mainVideo();//视频特征点检测  if (PROCEDURE == 3) return mainMatch();//图像与视频匹配  if (PROCEDURE == 4) return mainMotionPoints();//动点检测  if (PROCEDURE == 5) return mainStaticMatch();//图像与图像匹配  if (PROCEDURE == 6) return mainKmeans();//特征点聚类-Kmeas算法}

源码下载：http://download.csdn.net/detail/u011469591/5994129