Python调用cv.matchTemplate进行图像匹配的异常问题求解

问题修复方案 #

核心错误点 #

你现有代码中调用cv.matchTemplate时错误传入了img[0]、match[0]，这两个参数实际是原图和模板的第一行像素，而非完整图像，这是问题2、3的根本诱因。 #

各问题具体解决方法 #

• 问题1：保留颜色特征实现彩色模板匹配
  cv.matchTemplate本身支持3通道彩色图像输入，仅要求待匹配原图和模板的通道数、数据类型完全一致即可，无需强制转灰度。你可以直接删除两处转灰度的代码，保持输入为3通道BGR格式就能保留颜色特征完成匹配。
  额外注意：你用cv.IMREAD_UNCHANGED读取模板图时如果模板包含alpha透明通道，需要先将4通道转为3通道BGR，避免通道数不匹配报错。
• 问题2：匹配结果展示异常
  将cv.matchTemplate(img[0], match[0], cv.TM_CCOEFF_NORMED)修改为cv.matchTemplate(img, match, cv.TM_CCOEFF_NORMED)，传入完整的图像而非第一行像素，cv.imshow就会输出正常的匹配热力图。
• 问题3：置信度恒为1.0、位置固定为(0,0)
  该问题同样由传入单行像素导致：单行模板匹配单行原图时仅存在(0,0)这一个匹配位置，因此置信度永远为1.0。修复matchTemplate的传入参数后即可解决。

修正后可运行代码示例 #

import pyautogui
import numpy as np
import cv2 as cv
import time

while 1:
    # 读取屏幕截图，转为和模板一致的BGR3通道格式
    img = pyautogui.screenshot()
    img = np.array(img)
    img = cv.cvtColor(img, cv.COLOR_RGB2BGR)

    # 读取模板图，统一为3通道BGR格式
    match = cv.imread('match.png', cv.IMREAD_UNCHANGED)
    if match.shape[-1] == 4:
        match = cv.cvtColor(match, cv.COLOR_BGRA2BGR)

    # 执行模板匹配
    result = cv.matchTemplate(img, match, cv.TM_CCOEFF_NORMED)
    cv.imshow('bla', result)

    # 输出匹配结果
    min_val, max_val, min_loc, max_loc = cv.minMaxLoc(result)
    print('Best match top left position: %s' % str(max_loc))
    print('Best match confidence: %s' % max_val)

    if cv.waitKey(1) == ord('q'):
        cv.destroyAllWindows()
        break
    time.sleep(0.2)
print('done')

内容的提问来源于stack exchange，提问作者Radim Guichen