Python openCV在图像文件夹中查找图像(道路标志识别)

时间:2016-09-20 18:39:29

标签: python algorithm opencv template-matching

我正在进行道路标志识别计划。我成功完成了图像预处理。我提取了路标的图像。

现在我不知道应该使用什么算法或模板匹配来查找我提取的图像的匹配图像。

解决方案必须简单有效,因为我还在学习Python。

1.image:提取的图像2.图像:匹配的图像

Extracted image That image sould find algorithm in folder of images

1 个答案:

答案 0 :(得分:0)

模板匹配在您的情况下不起作用。模板匹配仅适用于源中几乎相同的模板。 我建议使用dlib http://dlib.net/ml.html深入学习。 它易于学习,您不需要编写大量代码。如果您仍想使用模板匹配,可以尝试以下操作:

import cv2

source = cv2.imread("/source/of/your_stop_sign")
template = cv2.imread("our/template")
(tempH, tempW) = template.shape[:2]

尝试找到模板:

result = cv2.matchTemplate(source, template, cv2.TM_CCOEFF)
(minVal, maxVal, minLoc, (x, y)) = cv2.minMaxLoc(result)

绘制边界框

cv2.rectangle(source, (x, y), (x + tempW, y + tempH), (0, 255, 0), 2)

显示结果

cv2.imshow("source", source)
cv2.imshow("template", template)
cv2.waitKey(0)

Template Source Result