117 lines
4.0 KiB
C++
117 lines
4.0 KiB
C++
|
// This file is part of OpenCV project.
|
||
|
|
// It is subject to the license terms in the LICENSE file found in the top-level directory
|
||
|
|
// of this distribution and at http://opencv.org/license.html.
|
||
|
|
|
||
|
|
#include "test_precomp.hpp"
|
||
|
|
|
||
|
|
namespace opencv_test { namespace {
|
||
|
|
|
||
|
|
class CV_GMSMatcherTest : public cvtest::BaseTest
|
||
|
|
{
|
||
|
|
public:
|
||
|
|
CV_GMSMatcherTest();
|
||
|
|
~CV_GMSMatcherTest();
|
||
|
|
|
||
|
|
protected:
|
||
|
|
virtual void run(int);
|
||
|
|
|
||
|
|
bool combinations[4][2];
|
||
|
|
double eps[3][4]; //3 imgs x 4 combinations
|
||
|
|
double correctMatchDistThreshold;
|
||
|
|
};
|
||
|
|
|
||
|
|
CV_GMSMatcherTest::CV_GMSMatcherTest()
|
||
|
|
{
|
||
|
|
combinations[0][0] = false; combinations[0][1] = false;
|
||
|
|
combinations[1][0] = false; combinations[1][1] = true;
|
||
|
|
combinations[2][0] = true; combinations[2][1] = false;
|
||
|
|
combinations[3][0] = true; combinations[3][1] = true;
|
||
|
|
|
||
|
|
eps[0][0] = 0.91;
|
||
|
|
eps[0][1] = 0.91;
|
||
|
|
eps[0][2] = 0.91;
|
||
|
|
eps[0][3] = 0.91;
|
||
|
|
|
||
|
|
eps[1][0] = 0.80;
|
||
|
|
eps[1][1] = 0.78;
|
||
|
|
eps[1][2] = 0.80;
|
||
|
|
eps[1][3] = 0.78;
|
||
|
|
|
||
|
|
eps[2][0] = 0.6;
|
||
|
|
eps[2][1] = 0.6;
|
||
|
|
eps[2][2] = 0.6;
|
||
|
|
eps[2][3] = 0.6;
|
||
|
|
|
||
|
|
correctMatchDistThreshold = 5.0;
|
||
|
|
}
|
||
|
|
|
||
|
|
CV_GMSMatcherTest::~CV_GMSMatcherTest() {}
|
||
|
|
|
||
|
|
void CV_GMSMatcherTest::run( int )
|
||
|
|
{
|
||
|
|
ts->set_failed_test_info(cvtest::TS::OK);
|
||
|
|
|
||
|
|
Mat imgRef = imread(string(ts->get_data_path()) + "detectors_descriptors_evaluation/images_datasets/graf/img1.png");
|
||
|
|
|
||
|
|
Ptr<Feature2D> orb = ORB::create(10000);
|
||
|
|
vector<KeyPoint> keypointsRef, keypointsCur;
|
||
|
|
Mat descriptorsRef, descriptorsCur;
|
||
|
|
orb->detectAndCompute(imgRef, noArray(), keypointsRef, descriptorsRef);
|
||
|
|
|
||
|
|
vector<DMatch> matchesAll, matchesGMS;
|
||
|
|
Ptr<DescriptorMatcher> matcher = DescriptorMatcher::create("BruteForce-Hamming");
|
||
|
|
|
||
|
|
const int startImg = 2;
|
||
|
|
const int nImgs = 3;
|
||
|
|
for (int num = startImg; num < startImg+nImgs; num++)
|
||
|
|
{
|
||
|
|
string fileName = cv::format("img%d.png", num);
|
||
|
|
string imgPath = string(ts->get_data_path()) + "detectors_descriptors_evaluation/images_datasets/graf/" + fileName;
|
||
|
|
Mat imgCur = imread(imgPath);
|
||
|
|
orb->detectAndCompute(imgCur, noArray(), keypointsCur, descriptorsCur);
|
||
|
|
|
||
|
|
matcher->match(descriptorsCur, descriptorsRef, matchesAll);
|
||
|
|
|
||
|
|
string xml = string(ts->get_data_path()) + format("detectors_descriptors_evaluation/images_datasets/graf/H1to%dp.xml", num);
|
||
|
|
FileStorage fs(xml, FileStorage::READ);
|
||
|
|
if (!fs.isOpened())
|
||
|
|
{
|
||
|
|
ts->set_failed_test_info(cvtest::TS::FAIL_INVALID_TEST_DATA);
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
|
||
|
|
Mat H1toCur;
|
||
|
|
fs[format("H1%d", num)] >> H1toCur;
|
||
|
|
|
||
|
|
for (int comb = 0; comb < 4; comb++)
|
||
|
|
{
|
||
|
|
matchGMS(imgCur.size(), imgRef.size(), keypointsCur, keypointsRef, matchesAll, matchesGMS, combinations[comb][0], combinations[comb][1]);
|
||
|
|
|
||
|
|
int nbCorrectMatches = 0;
|
||
|
|
for (size_t i = 0; i < matchesGMS.size(); i++)
|
||
|
|
{
|
||
|
|
Point2f ptRef = keypointsRef[matchesGMS[i].trainIdx].pt;
|
||
|
|
Point2f ptCur = keypointsCur[matchesGMS[i].queryIdx].pt;
|
||
|
|
Mat matRef = (Mat_<double>(3,1) << ptRef.x, ptRef.y, 1);
|
||
|
|
Mat matTrans = H1toCur * matRef;
|
||
|
|
Point2f ptTrans( (float) (matTrans.at<double>(0,0)/matTrans.at<double>(2,0)),
|
||
|
|
(float) (matTrans.at<double>(1,0)/matTrans.at<double>(2,0)));
|
||
|
|
|
||
|
|
if (cv::norm(ptTrans-ptCur) < correctMatchDistThreshold)
|
||
|
|
nbCorrectMatches++;
|
||
|
|
}
|
||
|
|
|
||
|
|
double ratio = nbCorrectMatches / (double) matchesGMS.size();
|
||
|
|
EXPECT_GT(ratio, eps[num-startImg][comb]) <<
|
||
|
|
cv::format("Invalid accuracy for image %s and combination withRotation=%d withScale=%d, "
|
||
|
|
"matches ratio is %g, ratio threshold is %g, distance threshold is %g.",
|
||
|
|
fileName.c_str(), combinations[comb][0], combinations[comb][1], ratio,
|
||
|
|
eps[num-startImg][comb], correctMatchDistThreshold);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
TEST(XFeatures2d_GMSMatcher, gms_matcher_regression) { CV_GMSMatcherTest test; test.safe_run(); }
|
||
|
|
|
||
|
|
}} // namespace
|