和前面利用暴力法找距离最近的descriptor,Flann-based matcher使用快速近似最近邻搜索算法,在匹配前可以利用图片训练该matcher,从而加快检测速度(TODO).
What is it Flann? FLANN is a library for performing fast approximate nearest neighbor searches in high dimensional spaces. It contains a collection of algorithms we found to work best for nearest neighbor search and a system for automatically choosing the
best algorithm and optimum parameters depending on the dataset.
#include <opencv2/core/core.hpp> #include <opencv2/highgui/highgui.hpp> #include <opencv2/nonfree/features2d.hpp> // #include <iostream> using namespace cv; using namespace std; int main(int argc, const char *argv[]){ Mat car1 = imread("car1.jpeg", 0);// load as grayscale Mat car2 = imread("car2.jpeg", 0); //cv::resize(car1,car1,car2.size()); SiftFeatureDetector detector; vector<KeyPoint> keypoints1, keypoints2; detector.detect(car1, keypoints1); detector.detect(car2, keypoints2); cout << "# keypoints of car1 :" << keypoints1.size() << endl; cout << "# keypoints of car2 :" << keypoints2.size() << endl; Mat descriptors1,descriptors2; Ptr<DescriptorExtractor> extractor = DescriptorExtractor::create("SIFT"); extractor->compute(car1,keypoints1,descriptors1); extractor->compute(car2,keypoints2,descriptors2); FlannBasedMatcher matcher; vector<DMatch> matches; matcher.match(descriptors1, descriptors2, matches); cout << "# matches : " << matches.size() << endl; //calc max and min keypoints distance double maxdist=0, mindist=100; for(int i=0;i<descriptors1.rows;i++){ double dist = matches[i].distance; //cout << "test" << dist << endl; if(dist<mindist) mindist = dist; if(dist > maxdist) maxdist = dist; } cout << "Max distance:" << maxdist << "Min distance:" << mindist << endl; // Get good matches (? distance is lesss than 2*min) vector<DMatch> goods; for(int i=0;i<descriptors1.rows;i++){ if(matches[i].distance < 2*mindist) goods.push_back(matches[i]); } cout << "# good matches : " << goods.size() << endl; // show it on an image Mat output; drawMatches(car1, keypoints1, car2, keypoints2, goods, output, Scalar::all(-1), Scalar::all(-1), vector<char>(), DrawMatchesFlags::NOT_DRAW_SINGLE_POINTS); imshow("good matches result",output); waitKey(0); return 0; }
运行结果,可看到经过滤后匹配数目大大减少。
参考:
1. Flann主页 http://www.cs.ubc.ca/research/flann/
2. Fast Approximate Nearest Neighbor Search
时间: 2024-10-19 01:01:18