python opencv matchTemplate是实现了掩码功能吗？

Question

OpenCV,从3.0.0版开始,为matchTemplate方法添加了一个掩码功能.它通过在模板上定义掩码来支持与透明模板的模板匹配.下面我的python程序工作正常,但如果我mask在cv2.matchTemplate调用中添加一个参数,它会抛出一个错误:

OpenCV Error: The function/feature is not implemented () in matchTemplateMask, file /Users/jared.rada/dev/opencv/modules/imgproc/src/templmatch.cpp, line 894
Traceback (most recent call last):
File "masked.py", line 13, in <module>
res = cv2.matchTemplate(img, tmpl, cv2.TM_CCOEFF_NORMED, data, mask)
cv2.error: /Users/jared.rada/dev/opencv/modules/imgproc/src/templmatch.cpp:894: error: (-213)  in function matchTemplateMask`

我的源代码:

import sys
import numpy as np
import cv2


img = cv2.imread('./image.jpg')
tmpl = cv2.imread('./tmpl.png')
mask = cv2.imread('./mask.png')
w, h = tmpl.shape[:-1]
data = np.zeros((h, w, 3), dtype=np.uint8)

res = cv2.matchTemplate(img, tmpl, cv2.TM_CCOEFF_NORMED, data, mask)

min_val, max_val, min_loc, max_loc = cv2.minMaxLoc(res)

top_left = max_loc
bottom_right = (top_left[0] + w, top_left[1] + h)
cv2.rectangle(img, top_left, bottom_right, (0, 0, 255), 2)

cv2.imshow("images", np.hstack([img]))
cv2.waitKey(0)

我怎么知道python绑定是否支持掩码功能？

Answer 1

有一个简单的答案:看SRC代码,你会发现,它的唯一的实现method == CV_TM_SQDIFF和method == CV_TM_CCORR_NORMED,换句话说,不是你想要的cv2.TM_CCOEFF_NORMED

Answer 2

更新答案：参见src代码

和提交：

templmatch：为所有方法添加对掩码的支持

现在最新的pip install opencv-contrib-python将包括所有方法的掩码（在 python3.7 上进行测试（