图像处理 模板匹配

演示用局部图片匹配整张图片

VS2017 + OpenCV4.0.1编译

#include <iostream>
#include <opencv2/opencv.hpp>
#include <opencv2/core.hpp>
#include <opencv2/highgui.hpp>
#include <opencv2/imgproc.hpp>
using namespace std;
using namespace cv;

//定义全局变量
Mat src, temp, res;

//显示窗口名
string img_window = "source image";
string res_window = "result image";

int match_method;
int max_tracker = 5;

//滑动条回调函数
void match(int pos, void*)
{
    Mat subImg;
    src.copyTo(subImg);//拷贝副本
    int res_row = src.rows - temp.rows + 1;
    int res_col = src.cols - temp.cols + 1;
    res.create(res_row, res_col, CV_32FC1);//创建结果矩阵

    //模板匹配
    matchTemplate(subImg, temp, res, match_method);
    normalize(res, res, 0, 1, NORM_MINMAX, -1, Mat());

    //定位
    double minVal, maxVal;
    Point minLoc, maxLoc, matchLoc;
    minMaxLoc(res, &minVal, &maxVal, &minLoc, &maxLoc, Mat());

    //分类讨论
    if (match_method == TM_SQDIFF || match_method == TM_SQDIFF_NORMED)
        matchLoc = minLoc;
    else
        matchLoc = maxLoc;

    //矩形框显示匹配结果
    rectangle(subImg, matchLoc, Point(matchLoc.x + temp.cols, matchLoc.y + temp.rows), Scalar(0, 255, 0), 2);
    rectangle(res, matchLoc, Point(matchLoc.x + temp.cols, matchLoc.y + temp.rows), Scalar(0, 255, 0), 2);
    
    imshow(img_window, subImg);
    imshow(res_window, res);
}

int main()
{
    //读入原图和模板图
    src = imread("D:\trashBox\testIMG\memory.jpg");
    temp = imread("D:\trashBox\testIMG\temp.png");
    imshow("模板", temp);
    
    //创建显示窗口
    namedWindow(img_window);
    namedWindow(res_window);

    //创建滑动条
    match(2, 0);
    createTrackbar("匹配方法", img_window, &match_method, max_tracker, match);
    
    waitKey(0);
    return 0;
}

参考资料

【1】http://www.opencv.org.cn/opencvdoc/2.3.2/html/doc/tutorials/imgproc/histograms/template_matching/template_matching.html#template-matching