OpenCV图像匹配算法之surf
//surf.cpp #include "stdafx.h" #include <cv.hpp> #include <highgui.h> #include "utils.h" #include <iostream> using namespace std; void surf(char* path1, char* path2, INFO& info, bool show) { double t1,t2; t1=cvGetTickCount(); initModule_nonfree(); Mat img1, img2; img1=imread(path1,0); img2=imread(path2,0); if(img1.data==NULL) { cout<<"The image can not been loaded: "<<path1<<endl; system("pause"); exit(-1); } if(img2.data==NULL) { cout<<"The image can not been loaded: "<<path2<<endl; system("pause"); exit(-1); } int minHessian=200; SurfFeatureDetector surf_detector(minHessian) ; SurfDescriptorExtractor surf_descriptor ; vector<KeyPoint> kpts1_surf, kpts2_surf; Mat desc1_surf, desc2_surf; Ptr<cv::DescriptorMatcher> matcher_l2 = DescriptorMatcher::create("BruteForce"); //欧氏距离匹配 vector<vector<DMatch> > dmatches_surf; vector<Point2f> matches_surf, inliers_surf; surf_detector.detect(img1,kpts1_surf); surf_detector.detect(img2,kpts2_surf); info.n1 = kpts1_surf.size(); info.n2 = kpts2_surf.size(); surf_descriptor.compute(img1,kpts1_surf,desc1_surf); surf_descriptor.compute(img2,kpts2_surf,desc2_surf); matcher_l2->knnMatch(desc1_surf,desc2_surf,dmatches_surf,2); matches2points_nndr(kpts1_surf,kpts2_surf,dmatches_surf,matches_surf,DRATIO); info.m=matches_surf.size()/2; compute_inliers_ransac(matches_surf,inliers_surf,MIN_H_ERROR,false); info.rm=inliers_surf.size()/2; t2=cvGetTickCount(); info.t=(t2-t1)/1000000.0/cvGetTickFrequency(); Mat img1_rgb_surf = imread(path1,1); Mat img2_rgb_surf = imread(path2,1); Mat img_com_surf = Mat(Size(img1.cols*2,img1.rows),CV_8UC3); if(show == true) { draw_inliers(img1_rgb_surf,img2_rgb_surf,img_com_surf,inliers_surf,2); imshow("surf",img_com_surf); waitKey(0); } return; }
使用
INFO surf_info; surf(path1,path2,surf_info,false); showInfo(surf_info);