update ch24

Author: makelove

ch24-模板匹配/matchTemplate.py renamed to ch24-模板匹配/24.1-OpenCV中的模板匹配-matchTemplate.py

@@ -2,6 +2,17 @@
 __author__ = 'play4fun'
 """
 create time:15-10-24 下午5:46
+原理
+模板匹配是用来在一副大图中搜寻查找模版图像位置的方法。OpenCV 为 我们提供了函数 cv2.matchTemplate()。
+和 2D 卷积一样 它也是用模板图像在输入图像 大图 上滑动 并在每一个位置对模板图像和与其对应的 输入图像的子区域  比较。
+OpenCV 提供了几种不同的比较方法 细节 看 文档 。 
+返回的结果是一个灰度图像 每一个像素值 示了此区域与模板的匹配 程度。
+如果输入图像的大小是 WxH  
+模板的大小是 wxh   输出的结果 的大小就是 W-w+1 H-h+1 。
+当你得到这幅图之后 就可以使用函数 cv2.minMaxLoc() 来找到其中的最小值和最大值的位置了。
+第一个值为矩形左上角的点 位置 
+w h 为 moban 模板矩形的宽和 。
+这个矩形就是 找到的模板区域了。
 """
 
 import cv2
@@ -16,6 +27,7 @@
 # All the 6 methods for comparison in a list
 methods = ['cv2.TM_CCOEFF', 'cv2.TM_CCOEFF_NORMED', 'cv2.TM_CCORR',
            'cv2.TM_CCORR_NORMED', 'cv2.TM_SQDIFF', 'cv2.TM_SQDIFF_NORMED']
+
 for meth in methods:
     img = img2.copy()
 
@@ -41,5 +53,5 @@
     plt.title('Matching Result'), plt.xticks([]), plt.yticks([])
     plt.subplot(122), plt.imshow(img, cmap='gray')
     plt.title('Detected Point'), plt.xticks([]), plt.yticks([])
-    plt.suptitle('method: '+meth)
+    plt.suptitle('method: ' + meth)
     plt.show()

ch24-模板匹配/24.2-多对象的模板匹配-Multiple-Objects.py

@@ -0,0 +1,29 @@
+'''
+假如你的目标对 只在图像中出现了很多次怎么办呢
+函数 cv.imMaxLoc() 只会给出最大值和最小值。此时 我们就 使用阈值了。
+在下 的例子中我们 经典游戏 Mario 的一张截屏图片中找到其中的硬币
+
+'''
+
+import cv2
+import numpy as np
+
+# from matplotlib import pyplot as plt
+
+img_rgb = cv2.imread('../data/mario.png')
+img_gray = cv2.cvtColor(img_rgb, cv2.COLOR_BGR2GRAY)
+template = cv2.imread('../data/mario_coin.png', 0)
+w, h = template.shape[::-1]
+
+res = cv2.matchTemplate(img_gray, template, cv2.TM_CCOEFF_NORMED)
+threshold = 0.8
+loc = np.where(res >= threshold)
+print(len(loc))
+
+for pt in zip(*loc[::-1]):
+    cv2.rectangle(img_rgb, pt, (pt[0] + w, pt[1] + h), (0, 255, 0), 2)
+    print("rectangle 1")
+
+# cv2.imwrite('res.png',img_rgb)
+cv2.imshow("result", img_rgb)
+cv2.waitKey(0)