0
点赞
收藏
分享

微信扫一扫

OpenCV 模板匹配

所谓的模板匹配,就是在给定的图片中查找和模板最相似的区域,该算法的输入包括模板和图片,整个任务的思路就是按照滑窗的思路不断的移动模板图片,计算其与图像中对应区域的匹配度,最终将匹配度最高的区域选择为最终的结果。

实现流程:

  • 准备两幅图像:
    1.原图像(I):在这幅图中,找到与模板相匹配的区域
    2.模板(T):与原图像进行比对的图像块
    OpenCV 模板匹配_模板匹配
  • 滑动模板图像和原图像进行比对:
    OpenCV 模板匹配_最小值_02

将模板块每次移动一个像素 (从左往右,从上往下),在每一个位置,都计算与模板图像的相似程度。

  • 对于每一个位置将计算的相似结果保存在结果矩阵(R)中。如果输入图像的大小(WxH)且模板图像的大小(wxh),则输出矩阵R的大小为(W-w + 1,H-h + 1)将R显示为图像,如下图所示:
    OpenCV 模板匹配_模板匹配_03

获得上述图像后,查找最大值所在的位置,那么该位置对应的区域就被认为是最匹配的。对应的区域就是以该点为顶点,长宽和模板图像一样大小的矩阵。

 res = cv.matchTemplate(img,template,method)

  • img: 要进行模板匹配的图像
  • Template :模板
  • method:实现模板匹配的算法,主要有:
  • 1.平方差匹配(CV_TM_SQDIFF):利用模板与图像之间的平方差进行匹配,最好的匹配是0,匹配越差,匹配的值越大。
  • 2.相关匹配(CV_TM_CCORR):利用模板与图像间的乘法进行匹配,数值越大表示匹配程度较高,越小表示匹配效果差。
  • 3.利用相关系数匹配(CV_TM_CCOEFF):利用模板与图像间的相关系数匹配,1表示完美的匹配,-1表示最差的匹配。

完成匹配后,使用cv.minMaxLoc()方法查找最大值所在的位置即可。如果使用平方差作为比较方法,则最小值位置是最佳匹配位置。

import cv2 as cv
from matplotlib import pyplot as plt

# 1 图像和模板读取
img = cv.imread('./1.png')
template = cv.imread('./img.png')
h, w, l = template.shape

# 2 模板匹配
# 2.1 模板匹配
res = cv.matchTemplate(img, template, cv.TM_SQDIFF)

# 2.2 返回图像中最匹配的位置,确定左上角的坐标,并将匹配位置绘制在图像上
min_val, max_val, min_loc, max_loc = cv.minMaxLoc(res) # 使用cv.minMaxLoc()搜索最匹配的位置。
# 使用平方差时最小值为最佳匹配位置
top_left = min_loc
bottom_right = (top_left[0] + w, top_left[1] + h)
cv.rectangle(img, top_left, bottom_right, (0, 255, 0), 2)

# 3 图像显示
plt.imshow(img[:, :, ::-1])
plt.title('匹配结果'),
plt.xticks([]),
plt.yticks([])
plt.show()

匹配模版:
OpenCV 模板匹配_最小值_04

匹配结果:
OpenCV 模板匹配_模板匹配_05

举报

相关推荐

0 条评论