Python+OpenCV虹膜扫描仪开发：模板匹配报错求助

我来帮你排查这个问题——你在开发虹膜扫描仪时遇到的cv2.matchTemplate()报错，核心原因是源图像和模板图像的通道数不匹配，下面一步步给你分析和解决：

你的代码与报错 #

先把你的代码和完整的报错信息整理出来（根据报错开头推断完整内容）：

代码片段

import cv2
import numpy as np
img1 = cv2.imread('canny.jpg'); img2 = cv2.imread('frame1.jpg');
edges=cv2.Canny(img2,100,100)
w,h=edges.shape[::-1]
res = cv2.matchTemplate(img1 , edges, cv2.TM_CCOEFF_NORMED)
threshold = 0.8
loc = np.where(res >= threshold)
print loc

完整报错信息

Traceback (most recent call last):
File "E:/OpenCV Programs/threshold2img1.py", line 5, in <module>
    res = cv2.matchTemplate(img1 , edges, cv2.TM_CCOEFF_NORMED)
error: (-215:Assertion failed) (depth == CV_8U || depth == CV_32F) && type == _templ.type() && _img.dims() <= 2 in function 'cv::matchTemplate'

问题根源 #

OpenCV的matchTemplate()函数有个硬性要求：输入的源图像和模板图像必须具有完全相同的通道数和数据类型。而你的代码里：

• img1 = cv2.imread('canny.jpg') 默认读取的是3通道的彩色图像（BGR格式）
• edges = cv2.Canny(img2,100,100) 输出的是单通道的灰度图像（Canny边缘检测的结果始终是单通道）
  两者通道数不一致，直接触发了断言错误。

两种解决方案 #

根据你的需求选择其中一种即可：

方案1：将源图像转为灰度图（推荐，因为模板是边缘图，灰度匹配足够）

如果你的canny.jpg本身是灰度图，或者不需要颜色信息参与匹配，直接把img1转为单通道灰度图：

import cv2
import numpy as np
# 读取时直接指定灰度模式，或者读取后转灰度
img1 = cv2.imread('canny.jpg', cv2.IMREAD_GRAYSCALE)
img2 = cv2.imread('frame1.jpg')
# 调整Canny阈值，高低阈值相同会导致边缘提取效果差
edges = cv2.Canny(img2, 100, 200)
w, h = edges.shape[::-1]
# 现在两者都是单通道，可正常匹配
res = cv2.matchTemplate(img1, edges, cv2.TM_CCOEFF_NORMED)
threshold = 0.8
loc = np.where(res >= threshold)
# Python3中print是函数，需要加括号
print(loc)

方案2：将模板转为3通道彩色图（仅当必须用彩色匹配时使用）

如果你需要保留源图像的颜色信息，把单通道的边缘图转为3通道：

import cv2
import numpy as np
img1 = cv2.imread('canny.jpg')
img2 = cv2.imread('frame1.jpg')
edges = cv2.Canny(img2, 100, 200)
# 将单通道灰度图转为3通道BGR图
edges_color = cv2.cvtColor(edges, cv2.COLOR_GRAY2BGR)
# 3通道图像的shape是(高, 宽, 3)，所以取前两个维度作为模板尺寸
w, h = edges_color.shape[1], edges_color.shape[0]
res = cv2.matchTemplate(img1, edges_color, cv2.TM_CCOEFF_NORMED)
threshold = 0.8
loc = np.where(res >= threshold)
print(loc)

额外小提示 #

• Canny边缘检测的高低阈值建议设置为不同值（比如100和200），相同阈值会导致边缘提取效果大打折扣，影响后续匹配精度。
• 如果你用的是Python3，print loc这种写法会报错，必须改成print(loc)。

内容的提问来源于stack exchange，提问作者Tejas Shaha