opencv3.4.5实现GMS+ORB特征匹配

最新推荐文章于 2024-06-22 21:05:45 发布

ChenCong7375

最新推荐文章于 2024-06-22 21:05:45 发布

阅读量7.2k

点赞数 5

文章标签： opencv GMS 特征匹配

本文链接：https://blog.csdn.net/ChenCong7375/article/details/89361542

版权

环境：opencv3.4.5+opencv_contrib，vs2015；

GMS链接：

GMS: Grid-based Motion Statistics for Fast, Ultra-robust Feature Correspondence CVPR2017

c++ code: https://github.com/JiawangBian/GMS-Feature-Matcher

代码：

#include <iostream>
#include <opencv2/core.hpp>
#include <opencv2/highgui.hpp>
#include <opencv2/imgproc.hpp>
#include <opencv2/flann.hpp>
#include <opencv2/features2d.hpp>
#include <opencv2/xfeatures2d.hpp>


using namespace cv;
using namespace std;
using namespace cv::xfeatures2d;


int main() 
{
	Mat img_1 = imread("01.jpg");
	Mat img_2 = imread("02.jpg");
	//Ptr<Feature2D> sift = xfeatures2d::SIFT::create();
	//Ptr<Feature2D> surf = xfeatures2d::SURF::create();
	Ptr<ORB> orb = ORB::create(5000);
	orb->setFastThreshold(0);

	vector<KeyPoint> keypoints_1, keypoints_2;
	Mat descriptors_1,descriptors_2;

	//sift->detect(img_1, keypoints_1);
	//sift->compute(img_1, keypoints_1, descriptors);

	orb->detectAndCompute(img_1, Mat(),keypoints_1, descriptors_1);
	orb->detectAndCompute(img_2, Mat(), keypoints_2, descriptors_2);

	//orb ->detect(img_1, keypoints_1);
	//orb ->compute(img_1, keypoints_1, descriptors_1);

	//orb->detect(img_2, keypoints_2);
	//orb->compute(img_2, keypoints_2, descriptors_2);

	Mat ShowKeypoints1, ShowKeypoints2;
	drawKeypoints(img_1, keypoints_1, ShowKeypoints1);
	drawKeypoints(img_2, keypoints_2, ShowKeypoints2);
	imshow("Result_1", ShowKeypoints1);
	imshow("Result_2", ShowKeypoints2);

    vector<DMatch> matchesAll, matchesGMS;
	BFMatcher matcher(NORM_HAMMING);
	
	matcher.match(descriptors_1, descriptors_2, matchesAll);
	cout << "matchesAll: " << matchesAll.size() << endl;
	matchGMS(img_1.size(), img_2.size(), keypoints_1, keypoints_2, matchesAll, matchesGMS);
	std::cout << "matchesGMS: " << matchesGMS.size() << std::endl;

	Mat finalMatches;
	drawMatches(img_1, keypoints_1, img_2, keypoints_2, matchesGMS, finalMatches, Scalar::all(-1), Scalar::all(-1),
		std::vector<char>(), DrawMatchesFlags::NOT_DRAW_SINGLE_POINTS);
	imshow("Matches GMS", finalMatches);
	imwrite("MatchesGMS.jpg", finalMatches);
	waitKey(0);

	return 0;
}

匹配结果：