A股上市公司传智教育(股票代码 003032)旗下技术交流社区北京昌平校区

 找回密码
 加入黑马

QQ登录

只需一步,快速开始


#include <iostream>
#include <opencv2/opencv.hpp>
#include"opencv2/xfeatures2d.hpp"


using namespace cv;
using namespace std;
using namespace cv::xfeatures2d;
int main()
{
        Mat a = imread("C:/Users/zhang/Desktop/77.png", IMREAD_GRAYSCALE);    //读取灰度图像
        Mat b = imread("C:/Users/zhang/Desktop/76.png", IMREAD_GRAYSCALE);

        Ptr<SURF> surf;
        surf = SURF::create(800);

        BFMatcher matcher;
        Mat c, d;
        vector<KeyPoint>key1, key2;
        vector<DMatch> matches;

        surf->detectAndCompute(a, Mat(), key1, c);
        surf->detectAndCompute(b, Mat(), key2, d);

        matcher.match(c, d, matches);       //匹配

        sort(matches.begin(), matches.end());  //筛选匹配点
        vector< DMatch > good_matches;
        int ptsPairs = std::min(50, (int)(matches.size() * 0.15));
        cout << ptsPairs << endl;
        for (int i = 0; i < ptsPairs; i++)
        {
                good_matches.push_back(matches[i]);
        }


        Mat outimg;
        drawMatches(a, key1, b, key2, good_matches, outimg, Scalar::all(-1), Scalar::all(-1), vector<char>(), DrawMatchesFlags::NOT_DRAW_SINGLE_POINTS);  //绘制匹配点

        vector<Point2f> obj;
        vector<Point2f> scene;

        for (size_t i = 0; i < good_matches.size(); i++)
        {
                obj.push_back(key1[good_matches[i].queryIdx].pt);
                //good_matches[i].queryIdx为好的匹配点的查询图像索引,如1,2,3
                //key1[good_matches[i].queryIdx].pt为查询图像上该特征点的坐标
                scene.push_back(key2[good_matches[i].trainIdx].pt);
                //good_matches[i].queryIdx为好的匹配点的训练图像索引,如1,2,3
                //key1[good_matches[i].queryIdx].pt为训练图像上该特征点的坐标
        }

        vector<Point2f> obj_corners(4);
        obj_corners[0] = Point(0, 0);
        obj_corners[1] = Point(a.cols, 0);
        obj_corners[2] = Point(a.cols, a.rows);
        obj_corners[3] = Point(0, a.rows);
        vector<Point2f> scene_corners(4);

        //透视变换
        Mat H = findHomography(obj, scene, RANSAC);
        //根据两幅图像的特征点求变换矩阵
        perspectiveTransform(obj_corners, scene_corners, H);
        //根据查询图像的坐标求训练图像ROI区域的坐标
        line(outimg, scene_corners[0] + Point2f((float)a.cols, 0), scene_corners[1] + Point2f((float)a.cols, 0), Scalar(0, 255, 0), 2, LINE_AA);       //绘制
        line(outimg, scene_corners[1] + Point2f((float)a.cols, 0), scene_corners[2] + Point2f((float)a.cols, 0), Scalar(0, 255, 0), 2, LINE_AA);
        line(outimg, scene_corners[2] + Point2f((float)a.cols, 0), scene_corners[3] + Point2f((float)a.cols, 0), Scalar(0, 255, 0), 2, LINE_AA);
        line(outimg, scene_corners[3] + Point2f((float)a.cols, 0), scene_corners[0] + Point2f((float)a.cols, 0), Scalar(0, 255, 0), 2, LINE_AA);
        imshow("aaaa", outimg);
        waitKey(0);
        return 0;
}



1 个回复

倒序浏览
奈斯,感谢分享
回复 使用道具 举报
您需要登录后才可以回帖 登录 | 加入黑马