opencv/doc/tutorials/imgproc/histograms/template_matching/template_matching.rst

373 lines
12 KiB
ReStructuredText
Raw Normal View History

.. _template_matching:
Template Matching
*****************
2011-07-06 16:15:13 +02:00
Goal
====
In this tutorial you will learn how to:
.. container:: enumeratevisibleitemswithsquare
* Use the OpenCV function :match_template:`matchTemplate <>` to search for matches between an image patch and an input image
* Use the OpenCV function :min_max_loc:`minMaxLoc <>` to find the maximum and minimum values (as well as their positions) in a given array.
Theory
======
What is template matching?
--------------------------
.. container:: enumeratevisibleitemswithsquare
Template matching is a technique for finding areas of an image that match (are similar) to a template image (patch).
How does it work?
------------------
.. container:: enumeratevisibleitemswithsquare
* We need two primary components:
a. **Source image (I):** The image in which we expect to find a match to the template image
b. **Template image (T):** The patch image which will be compared to the template image
2011-07-06 16:15:13 +02:00
our goal is to detect the highest matching area:
.. image:: images/Template_Matching_Template_Theory_Summary.jpg
:align: center
2011-07-06 16:15:13 +02:00
* To identify the matching area, we have to *compare* the template image against the source image by sliding it:
.. image:: images/Template_Matching_Template_Theory_Sliding.jpg
:align: center
2011-07-06 16:15:13 +02:00
* By **sliding**, we mean moving the patch one pixel at a time (left to right, up to down). At each location, a metric is calculated so it represents how "good" or "bad" the match at that location is (or how similar the patch is to that particular area of the source image).
* For each location of **T** over **I**, you *store* the metric in the *result matrix* **(R)**. Each location :math:`(x,y)` in **R** contains the match metric:
.. image:: images/Template_Matching_Template_Theory_Result.jpg
:align: center
2011-07-06 16:15:13 +02:00
the image above is the result **R** of sliding the patch with a metric **TM_CCORR_NORMED**. The brightest locations indicate the highest matches. As you can see, the location marked by the red circle is probably the one with the highest value, so that location (the rectangle formed by that point as a corner and width and height equal to the patch image) is considered the match.
2011-07-06 16:15:13 +02:00
* In practice, we use the function :min_max_loc:`minMaxLoc <>` to locate the highest value (or lower, depending of the type of matching method) in the *R* matrix.
2011-07-06 16:15:13 +02:00
Which are the matching methods available in OpenCV?
----------------------------------------------------
Good question. OpenCV implements Template matching in the function :match_template:`matchTemplate <>`. The available methods are 6:
a. **method=CV\_TM\_SQDIFF**
2011-07-06 16:15:13 +02:00
.. math::
R(x,y)= \sum _{x',y'} (T(x',y')-I(x+x',y+y'))^2
2011-07-06 16:15:13 +02:00
b. **method=CV\_TM\_SQDIFF\_NORMED**
2011-07-06 16:15:13 +02:00
.. math::
R(x,y)= \frac{\sum_{x',y'} (T(x',y')-I(x+x',y+y'))^2}{\sqrt{\sum_{x',y'}T(x',y')^2 \cdot \sum_{x',y'} I(x+x',y+y')^2}}
2011-07-06 16:15:13 +02:00
c. **method=CV\_TM\_CCORR**
2011-07-06 16:15:13 +02:00
.. math::
R(x,y)= \sum _{x',y'} (T(x',y') \cdot I(x+x',y+y'))
d. **method=CV\_TM\_CCORR\_NORMED**
2011-07-06 16:15:13 +02:00
.. math::
R(x,y)= \frac{\sum_{x',y'} (T(x',y') \cdot I'(x+x',y+y'))}{\sqrt{\sum_{x',y'}T(x',y')^2 \cdot \sum_{x',y'} I(x+x',y+y')^2}}
2011-07-06 16:15:13 +02:00
e. **method=CV\_TM\_CCOEFF**
2011-07-06 16:15:13 +02:00
.. math::
R(x,y)= \sum _{x',y'} (T'(x',y') \cdot I(x+x',y+y'))
2011-07-06 16:15:13 +02:00
where
2011-07-06 16:15:13 +02:00
.. math::
\begin{array}{l} T'(x',y')=T(x',y') - 1/(w \cdot h) \cdot \sum _{x'',y''} T(x'',y'') \\ I'(x+x',y+y')=I(x+x',y+y') - 1/(w \cdot h) \cdot \sum _{x'',y''} I(x+x'',y+y'') \end{array}
2011-07-06 16:15:13 +02:00
f. **method=CV\_TM\_CCOEFF\_NORMED**
2011-07-06 16:15:13 +02:00
.. math::
R(x,y)= \frac{ \sum_{x',y'} (T'(x',y') \cdot I'(x+x',y+y')) }{ \sqrt{\sum_{x',y'}T'(x',y')^2 \cdot \sum_{x',y'} I'(x+x',y+y')^2} }
2011-07-06 16:15:13 +02:00
Code
====
.. container:: enumeratevisibleitemswithsquare
* **What does this program do?**
2011-07-06 16:15:13 +02:00
.. container:: enumeratevisibleitemswithsquare
* Loads an input image and a image patch (*template*)
* Perform a template matching procedure by using the OpenCV function :match_template:`matchTemplate <>` with any of the 6 matching methods described before. The user can choose the method by entering its selection in the Trackbar.
* Normalize the output of the matching procedure
* Localize the location with higher matching probability
* Draw a rectangle around the area corresponding to the highest match
* **Downloadable code**:
Click `here <http://code.opencv.org/projects/opencv/repository/revisions/master/raw/samples/cpp/tutorial_code/Histograms_Matching/MatchTemplate_Demo.cpp>`_
2011-07-06 16:15:13 +02:00
* **Code at glance:**
.. code-block:: cpp
#include "opencv2/highgui/highgui.hpp"
2011-07-06 16:15:13 +02:00
#include "opencv2/imgproc/imgproc.hpp"
#include <iostream>
#include <stdio.h>
using namespace std;
using namespace cv;
/// Global Variables
Mat img; Mat templ; Mat result;
char* image_window = "Source Image";
char* result_window = "Result window";
int match_method;
int max_Trackbar = 5;
/// Function Headers
void MatchingMethod( int, void* );
/** @function main */
int main( int argc, char** argv )
{
/// Load image and template
img = imread( argv[1], 1 );
templ = imread( argv[2], 1 );
/// Create windows
namedWindow( image_window, CV_WINDOW_AUTOSIZE );
namedWindow( result_window, CV_WINDOW_AUTOSIZE );
2011-07-06 16:15:13 +02:00
/// Create Trackbar
char* trackbar_label = "Method: \n 0: SQDIFF \n 1: SQDIFF NORMED \n 2: TM CCORR \n 3: TM CCORR NORMED \n 4: TM COEFF \n 5: TM COEFF NORMED";
createTrackbar( trackbar_label, image_window, &match_method, max_Trackbar, MatchingMethod );
MatchingMethod( 0, 0 );
waitKey(0);
return 0;
}
/**
* @function MatchingMethod
* @brief Trackbar callback
*/
void MatchingMethod( int, void* )
{
/// Source image to display
Mat img_display;
img.copyTo( img_display );
2011-07-06 16:15:13 +02:00
/// Create the result matrix
int result_cols = img.cols - templ.cols + 1;
int result_rows = img.rows - templ.rows + 1;
2011-07-06 16:15:13 +02:00
result.create( result_cols, result_rows, CV_32FC1 );
/// Do the Matching and Normalize
matchTemplate( img, templ, result, match_method );
normalize( result, result, 0, 1, NORM_MINMAX, -1, Mat() );
/// Localizing the best match with minMaxLoc
double minVal; double maxVal; Point minLoc; Point maxLoc;
Point matchLoc;
2011-07-06 16:15:13 +02:00
minMaxLoc( result, &minVal, &maxVal, &minLoc, &maxLoc, Mat() );
/// For SQDIFF and SQDIFF_NORMED, the best matches are lower values. For all the other methods, the higher the better
if( match_method == CV_TM_SQDIFF || match_method == CV_TM_SQDIFF_NORMED )
{ matchLoc = minLoc; }
else
2011-07-06 16:15:13 +02:00
{ matchLoc = maxLoc; }
/// Show me what you got
rectangle( img_display, matchLoc, Point( matchLoc.x + templ.cols , matchLoc.y + templ.rows ), Scalar::all(0), 2, 8, 0 );
rectangle( result, matchLoc, Point( matchLoc.x + templ.cols , matchLoc.y + templ.rows ), Scalar::all(0), 2, 8, 0 );
2011-07-06 16:15:13 +02:00
imshow( image_window, img_display );
imshow( result_window, result );
return;
}
Explanation
===========
#. Declare some global variables, such as the image, template and result matrices, as well as the match method and the window names:
.. code-block:: cpp
Mat img; Mat templ; Mat result;
char* image_window = "Source Image";
char* result_window = "Result window";
int match_method;
int max_Trackbar = 5;
#. Load the source image and template:
.. code-block:: cpp
img = imread( argv[1], 1 );
templ = imread( argv[2], 1 );
#. Create the windows to show the results:
.. code-block:: cpp
namedWindow( image_window, CV_WINDOW_AUTOSIZE );
namedWindow( result_window, CV_WINDOW_AUTOSIZE );
2011-07-06 16:15:13 +02:00
#. Create the Trackbar to enter the kind of matching method to be used. When a change is detected the callback function **MatchingMethod** is called.
.. code-block:: cpp
char* trackbar_label = "Method: \n 0: SQDIFF \n 1: SQDIFF NORMED \n 2: TM CCORR \n 3: TM CCORR NORMED \n 4: TM COEFF \n 5: TM COEFF NORMED";
createTrackbar( trackbar_label, image_window, &match_method, max_Trackbar, MatchingMethod );
#. Wait until user exits the program.
.. code-block:: cpp
waitKey(0);
return 0;
2011-07-06 16:15:13 +02:00
#. Let's check out the callback function. First, it makes a copy of the source image:
.. code-block:: cpp
Mat img_display;
img.copyTo( img_display );
#. Next, it creates the result matrix that will store the matching results for each template location. Observe in detail the size of the result matrix (which matches all possible locations for it)
.. code-block:: cpp
2011-07-06 16:15:13 +02:00
int result_cols = img.cols - templ.cols + 1;
int result_rows = img.rows - templ.rows + 1;
2011-07-06 16:15:13 +02:00
result.create( result_cols, result_rows, CV_32FC1 );
2011-07-06 16:15:13 +02:00
#. Perform the template matching operation:
.. code-block:: cpp
matchTemplate( img, templ, result, match_method );
the arguments are naturally the input image **I**, the template **T**, the result **R** and the match_method (given by the Trackbar)
#. We normalize the results:
.. code-block:: cpp
normalize( result, result, 0, 1, NORM_MINMAX, -1, Mat() );
#. We localize the minimum and maximum values in the result matrix **R** by using :min_max_loc:`minMaxLoc <>`.
2011-07-06 16:15:13 +02:00
.. code-block:: cpp
double minVal; double maxVal; Point minLoc; Point maxLoc;
Point matchLoc;
2011-07-06 16:15:13 +02:00
minMaxLoc( result, &minVal, &maxVal, &minLoc, &maxLoc, Mat() );
2011-07-06 16:15:13 +02:00
the function calls as arguments:
.. container:: enumeratevisibleitemswithsquare
2011-07-06 16:15:13 +02:00
+ **result:** The source array
+ **&minVal** and **&maxVal:** Variables to save the minimum and maximum values in **result**
+ **&minLoc** and **&maxLoc:** The Point locations of the minimum and maximum values in the array.
+ **Mat():** Optional mask
#. For the first two methods ( CV\_SQDIFF and CV\_SQDIFF\_NORMED ) the best match are the lowest values. For all the others, higher values represent better matches. So, we save the corresponding value in the **matchLoc** variable:
.. code-block:: cpp
2011-07-06 16:15:13 +02:00
if( match_method == CV_TM_SQDIFF || match_method == CV_TM_SQDIFF_NORMED )
{ matchLoc = minLoc; }
else
2011-07-06 16:15:13 +02:00
{ matchLoc = maxLoc; }
#. Display the source image and the result matrix. Draw a rectangle around the highest possible matching area:
.. code-block:: cpp
rectangle( img_display, matchLoc, Point( matchLoc.x + templ.cols , matchLoc.y + templ.rows ), Scalar::all(0), 2, 8, 0 );
rectangle( result, matchLoc, Point( matchLoc.x + templ.cols , matchLoc.y + templ.rows ), Scalar::all(0), 2, 8, 0 );
2011-07-06 16:15:13 +02:00
imshow( image_window, img_display );
imshow( result_window, result );
Results
=======
#. Testing our program with an input image such as:
.. image:: images/Template_Matching_Original_Image.jpg
:align: center
2011-07-06 16:15:13 +02:00
and a template image:
.. image:: images/Template_Matching_Template_Image.jpg
:align: center
2011-07-06 16:15:13 +02:00
#. Generate the following result matrices (first row are the standard methods SQDIFF, CCORR and CCOEFF, second row are the same methods in its normalized version). In the first column, the darkest is the better match, for the other two columns, the brighter a location, the higher the match.
============ ============ ============
2011-07-06 16:15:13 +02:00
|Result_0| |Result_2| |Result_4|
============ ============ ============
2011-07-06 16:15:13 +02:00
|Result_1| |Result_3| |Result_5|
============ ============ ============
2011-07-06 16:15:13 +02:00
.. |Result_0| image:: images/Template_Matching_Correl_Result_0.jpg
:align: middle
.. |Result_1| image:: images/Template_Matching_Correl_Result_1.jpg
:align: middle
.. |Result_2| image:: images/Template_Matching_Correl_Result_2.jpg
:align: middle
.. |Result_3| image:: images/Template_Matching_Correl_Result_3.jpg
:align: middle
.. |Result_4| image:: images/Template_Matching_Correl_Result_4.jpg
:align: middle
.. |Result_5| image:: images/Template_Matching_Correl_Result_5.jpg
:align: middle
#. The right match is shown below (black rectangle around the face of the guy at the right). Notice that CCORR and CCDEFF gave erroneous best matches, however their normalized version did it right, this may be due to the fact that we are only considering the "highest match" and not the other possible high matches.
.. image:: images/Template_Matching_Image_Result.jpg
:align: center