欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

ORB特征点检测和匹配

程序员文章站 2022-06-11 11:20:11
...
#include <stdio.h>
#include <iostream>
#include "opencv2/core/core.hpp"
#include "opencv2/features2d/features2d.hpp"
#include "opencv2/highgui/highgui.hpp"
#include <iostream>
#include <vector>
#include <QtDebug>

using namespace cv;
using namespace std;

char* image_filename2 = "F:/opencv-3.1.0/pictures/test2.bmp";
char* image_filename1 = "F:/opencv-3.1.0/pictures/test1.bmp";


int main( int argc, char** argv )
{

    Mat img_1 = imread(image_filename2);
    Mat img_2 = imread(image_filename1);
    if (!img_1.data || !img_2.data)
    {
        cout << "error reading images " << endl;
        return -1;
    }

    //ORB orb;
    Ptr<ORB> orb = ORB::create(500,1.2f,8,31,0,2,ORB::HARRIS_SCORE,31,20);
    vector<KeyPoint> keyPoints_1, keyPoints_2;
    Mat descriptors_1, descriptors_2;

#if 0
    orb->detect(img_1,keyPoints_1, Mat());
    for(int i=0;i<keyPoints_1.size();i++)
    {
        qDebug("keypoint[%d] x:%f y:%f size:%d angle:%f\n", i, keyPoints_1[i].pt.x, keyPoints_1[i].pt.y, keyPoints_1[i].size, keyPoints_1[i].angle);
    }
#endif

    orb->detectAndCompute(img_1, Mat(), keyPoints_1, descriptors_1);
    orb->detectAndCompute(img_2, Mat(), keyPoints_2, descriptors_2);

    qDebug("keyPoints1 size:%d\n", keyPoints_1.size());
    qDebug("keyPoints2 size:%d\n", keyPoints_2.size());

    flann::Index flannIndex(descriptors_2,flann::LshIndexParams(20,10,2),cvflann::FLANN_DIST_HAMMING);

    Mat matchIndex(descriptors_1.rows,2,CV_32SC1),
                matchDistance(descriptors_1.rows,2,CV_32FC1);

    flannIndex.knnSearch(descriptors_1,matchIndex,matchDistance,2,flann::SearchParams());

    int goodMatchsNum=0;
    std::vector< DMatch > good_matches;
    for(int i=0;i<matchDistance.rows;i++)
    {
       if (matchDistance.at<float>(i,0)<0.6*matchDistance.at<float>(i,1))
       {
           DMatch dmatches(i,matchIndex.at<int>(i,0),
               matchDistance.at<float>(i,0));
           good_matches.push_back( dmatches);
           goodMatchsNum++;

       }
    }
    cout<<goodMatchsNum<<endl;

    //-- Draw only "good" matches
    Mat img_matches;
    drawMatches( img_1, keyPoints_1, img_2, keyPoints_2,
               good_matches, img_matches, Scalar::all(-1), Scalar::all(-1),
               vector<char>(), DrawMatchesFlags::NOT_DRAW_SINGLE_POINTS );

    //-- Show detected matches
    imshow( "Good Matches", img_matches );

    for( int i = 0; i < good_matches.size(); i++ )
    { qDebug( "-- Good Match [%d] Keypoint 1: %d  -- Keypoint 2: %d  \n", i, good_matches[i].queryIdx, good_matches[i].trainIdx ); }

    waitKey(0);

    return 0;
 }
ORB特征点检测和匹配