如何减少模板匹配的图片搜索区域？

Question

我正在使用openCV模板匹配算法，但是在基础图像中搜索模板图像需要很长时间。 有什么方法可以告诉算法只搜索给定基本图像的特定区域（如顶部、底部、左侧、右侧）以获取模板图像？ 我的目标是减少搜索时间，任何其他方法将不胜感激。

这是我的代码 -

template = cv2.imread(temp_img)
template = cv2.cvtColor(template, cv2.COLOR_BGR2GRAY)
sigma=0.33
v = np.median(template)
lower = int(max(0, (1.0 - sigma) * v))
upper = int(min(255, (1.0 + sigma) * v))
template = cv2.Canny(template, lower, upper)
(tH, tW) = template.shape[:2]
baseImage = cv2.imread(base_img)
gray = cv2.cvtColor(baseImage, cv2.COLOR_BGR2GRAY)
found = None
threshold = 0.8
for scale in np.linspace(0.1, 0.5, 30)[::-1]:
    resized = imutils.resize(gray, width=int(gray.shape[1] * scale))
    r = gray.shape[1] / float(resized.shape[1])
    if resized.shape[0] < tH or resized.shape[1] < tW:
        break   
    v = np.median(resized)
    lower = int(max(0, (1.0 - sigma) * v))
    upper = int(min(255, (1.0 + sigma) * v))
    canny = cv2.Canny(resized, lower, upper)
    res = cv2.matchTemplate(canny, template, cv2.TM_CCOEFF_NORMED)
    (min_val, max_val, _, max_loc) = cv2.minMaxLoc(res) 
    if found is None or max_val > found[0]:
        found = (max_val, max_loc, r)               
(max_val, max_loc, r) = found
if max_val > threshold:
    (start_x, start_y) = (int(max_loc[0] * r), int(max_loc[1] * r))
    (end_x, end_y) = (int((max_loc[0] + tW) * r), int((max_loc[1] + tH) * r))
    cv2.rectangle(original_image, (start_x, start_y), (end_x, end_y), (0,255,0), 2)
    cv2.imshow('detected', original_image)
    cv2.imwrite('detected.png', original_image)
    cv2.waitKey(0)

Answer 1

需要很长时间的原因是您正在给定图像中搜索模板图像的每个比例。

for scale in np.linspace(0.1, 0.5, 30)[::-1]:
    resized = imutils.resize(gray, width=int(gray.shape[1] * scale))

您可以删除减少搜索时间的for循环。 权衡可能是缺少模板图像。 例如：如果给定图像中有 5 个模板图像，您可能会找到其中的 2 个。

第二个选项是更改linspace参数：

• 当前参数：

---

• 开始： 0.1

• 停止： 0.5

• 数量： 30

• 如果我们解释当前的参数：

---

• 从0.1到0.5比率范围：

  • 
  • 您正在从每个样本中生成 30 个样本。

• 可能的解决方案：

---

• 您可以缩短范围，即从 0.4-0.5 np.linspace(0.4, 0.5, 30)

• 或者你可以减少生成的样本大小，即 20, np.linspace(0.1, 0.5, 20)