diff --git a/modules/features2d/include/opencv2/features2d/features2d.hpp b/modules/features2d/include/opencv2/features2d/features2d.hpp
index 1c6d56d75..c54f1ca5f 100644
--- a/modules/features2d/include/opencv2/features2d/features2d.hpp
+++ b/modules/features2d/include/opencv2/features2d/features2d.hpp
@@ -1751,6 +1751,8 @@ public:
// Writes match object to a file storage
virtual void write( FileStorage& fs ) const {};
+ static GenericDescriptorMatch* CreateDescriptorMatch( const string &alg_name, const string ¶ms_filename = string () );
+
protected:
KeyPointCollection collection;
};
diff --git a/modules/features2d/src/descriptors.cpp b/modules/features2d/src/descriptors.cpp
index 4825f4a8b..8fd5a1a3d 100644
--- a/modules/features2d/src/descriptors.cpp
+++ b/modules/features2d/src/descriptors.cpp
@@ -226,6 +226,38 @@ void GenericDescriptorMatch::clear()
{
collection.clear();
}
+
+GenericDescriptorMatch* GenericDescriptorMatch::CreateDescriptorMatch( const string &alg_name, const string ¶ms_filename )
+{
+ GenericDescriptorMatch *descriptorMatch = 0;
+ if( ! alg_name.compare ("one_way") )
+ {
+ descriptorMatch = new OneWayDescriptorMatch ();
+ }
+ else if( ! alg_name.compare ("fern") )
+ {
+ FernDescriptorMatch::Params params;
+ params.signatureSize = INT_MAX;
+ descriptorMatch = new FernDescriptorMatch (params);
+ }
+ else if( ! alg_name.compare ("calonder") )
+ {
+ descriptorMatch = new CalonderDescriptorMatch ();
+ }
+
+ if( !params_filename.empty() && descriptorMatch != 0 )
+ {
+ FileStorage fs = FileStorage( params_filename, FileStorage::READ );
+ if( fs.isOpened() )
+ {
+ descriptorMatch->read( fs.root() );
+ fs.release();
+ }
+ }
+
+ return descriptorMatch;
+}
+
/****************************************************************************************\
* OneWayDescriptorMatch *
\****************************************************************************************/
diff --git a/samples/c/calonder_params.xml b/samples/c/calonder_params.xml
new file mode 100644
index 000000000..f2294caef
--- /dev/null
+++ b/samples/c/calonder_params.xml
@@ -0,0 +1,10 @@
+
+
+ 20
+ 7
+ 1000
+ 20
+ 30
+ 4
+ 1
+
diff --git a/samples/c/fern_params.xml b/samples/c/fern_params.xml
new file mode 100644
index 000000000..d083534e8
--- /dev/null
+++ b/samples/c/fern_params.xml
@@ -0,0 +1,10 @@
+
+
+ 0
+ 31
+ INT_MAX
+ 50
+ 9
+ 1000
+ 0
+
diff --git a/samples/c/match_sample.cpp b/samples/c/match_sample.cpp
new file mode 100644
index 000000000..863e4197b
--- /dev/null
+++ b/samples/c/match_sample.cpp
@@ -0,0 +1,94 @@
+#include
+#include
+#include
+
+#include
+
+using namespace cv;
+
+IplImage* DrawCorrespondences(IplImage* img1, const vector& features1, IplImage* img2,
+ const vector& features2, const vector& desc_idx);
+
+int main(int argc, char** argv)
+{
+ if (argc != 5)
+ {
+ printf("Format: \n./match_sample [image1] [image2] [algorithm] [XML params]\n");
+ printf("For example: ./match_sample scene_l.bmp scene_r.bmp fern fern_params.xml\n");
+ return 0;
+ }
+
+ std::string img1_name = std::string(argv[1]);
+ std::string img2_name = std::string(argv[2]);
+ std::string alg_name = std::string(argv[3]);
+ std::string params_filename = std::string(argv[4]);
+
+ GenericDescriptorMatch *descriptorMatch = GenericDescriptorMatch::CreateDescriptorMatch (alg_name, params_filename);
+ if( descriptorMatch == 0 )
+ {
+ printf ("Cannot create descriptor\n");
+ return 0;
+ }
+
+ //printf("Reading the images...\n");
+ IplImage* img1 = cvLoadImage(img1_name.c_str(), CV_LOAD_IMAGE_GRAYSCALE);
+ IplImage* img2 = cvLoadImage(img2_name.c_str(), CV_LOAD_IMAGE_GRAYSCALE);
+
+ // extract keypoints from the first image
+ SURF surf_extractor(5.0e3);
+ vector keypoints1;
+
+ // printf("Extracting keypoints\n");
+ surf_extractor(img1, Mat(), keypoints1);
+
+ printf("Extracted %d keypoints from the first image\n", (int)keypoints1.size());
+
+ vector keypoints2;
+ surf_extractor(img2, Mat(), keypoints2);
+ printf("Extracted %d keypoints from the second image\n", (int)keypoints2.size());
+
+ printf("Finding nearest neighbors... \n");
+ // find NN for each of keypoints2 in keypoints1
+ descriptorMatch->add( img1, keypoints1 );
+ vector matches2to1;
+ matches2to1.resize(keypoints2.size());
+ descriptorMatch->match( img2, keypoints2, matches2to1 );
+ printf("Done\n");
+
+ IplImage* img_corr = DrawCorrespondences(img1, keypoints1, img2, keypoints2, matches2to1);
+
+ cvNamedWindow("correspondences", 1);
+ cvShowImage("correspondences", img_corr);
+ cvWaitKey(0);
+
+ cvReleaseImage(&img1);
+ cvReleaseImage(&img2);
+ cvReleaseImage(&img_corr);
+ delete descriptorMatch;
+}
+
+IplImage* DrawCorrespondences(IplImage* img1, const vector& features1, IplImage* img2,
+ const vector& features2, const vector& desc_idx)
+{
+ IplImage* img_corr = cvCreateImage(cvSize(img1->width + img2->width, MAX(img1->height, img2->height)),
+ IPL_DEPTH_8U, 3);
+ cvSetImageROI(img_corr, cvRect(0, 0, img1->width, img1->height));
+ cvCvtColor(img1, img_corr, CV_GRAY2RGB);
+ cvSetImageROI(img_corr, cvRect(img1->width, 0, img2->width, img2->height));
+ cvCvtColor(img2, img_corr, CV_GRAY2RGB);
+ cvResetImageROI(img_corr);
+
+ for (size_t i = 0; i < features1.size(); i++)
+ {
+ cvCircle(img_corr, features1[i].pt, 3, CV_RGB(255, 0, 0));
+ }
+
+ for (size_t i = 0; i < features2.size(); i++)
+ {
+ CvPoint pt = cvPoint(features2[i].pt.x + img1->width, features2[i].pt.y);
+ cvCircle(img_corr, pt, 3, CV_RGB(255, 0, 0));
+ cvLine(img_corr, features1[desc_idx[i]].pt, pt, CV_RGB(0, 255, 0));
+ }
+
+ return img_corr;
+}