Python和OpenCV进行多尺度模板匹配实现

 # USAGE # python match.py --template cod_logo.png-600 --images images # USAGE2 了解实际检测原理及细节 # python match.py --template cod_logo.png-600 --images images --visualize 1 # 导入必要的包 import argparse  # argparse解析命令行参数 import glob  # 获取输入图像的路径 import cv2  # opencv绑定 import imutils  # 图像处理的一些方法 import numpy as np  # numpy进行数值处理 # 构建命令行及解析参数 # --template 模板路径 # --images 原始图像路径 # --visualize 标志是否显示每一个迭代的可视化结果 ap = argparse.ArgumentParser() ap.add_argument("-t", "--template", required=True, help="Path to template image") ap.add_argument("-i", "--images", required=True, help="Path to images where template will be matched") ap.add_argument("-v", "--visualize", help="Flag indicating whether or not to visualize each iteration") args = vars(ap.parse_args()) # 加载模板图像，转换灰度图，检测边缘 # 使用边缘而不是原始图像进行模板匹配可以大大提高模板匹配的精度。 template = cv2.imread(args["template"]) template = cv2.cvtColor(template, cv2.COLOR_BGR2GRAY) template = cv2.Canny(template, 50, 200) (tH, tW) = template.shape[:2] cv2.imshow("Template", template) # 遍历图像以匹配模板 for imagePath in glob.glob(args["images"] + "/*.jpg-600"): # 加载图像，转换为灰度图，初始化用于追踪匹配区域的簿记变量 image = cv2.imread(imagePath) gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY) found = None # 遍历图像尺寸 for scale in np.linspace(0.2, 1.0, 20)[::-1]: # 根据scale比例缩放图像，并保持其宽高比 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]  found[0]: found = (maxVal, maxLoc, r) # 解包簿记变量并<mark style="color:transparent">来源gaodaimacom搞#代%码网</mark>基于调整大小的比率，计算边界框（x，y）坐标 (_, maxLoc, r) = found (startX, startY) = (int(maxLoc[0] * r), int(maxLoc[1] * r)) (endX, endY) = (int((maxLoc[0] + tW) * r), int((maxLoc[1] + tH) * r)) # 在检测结果上绘制边界框并展示图像 cv2.rectangle(image, (startX, startY), (endX, endY), (0, 0, 255), 2) cv2.imshow("Image", image) cv2.waitKey(0)