added tutorial 7

longqua69 · Feb 6, 2021 · 941d2bf · 941d2bf
1 parent a0f9977
commit 941d2bf
Show file tree

Hide file tree

Showing 4 changed files with 25 additions and 0 deletions.
diff --git a/assets/ball.PNG b/assets/ball.PNG
diff --git a/assets/shoe.PNG b/assets/shoe.PNG
diff --git a/assets/soccer_practice.jpg b/assets/soccer_practice.jpg
diff --git a/tutorial7.py b/tutorial7.py
@@ -0,0 +1,25 @@
+import numpy as np
+import cv2
+
+img = cv2.resize(cv2.imread('assets/soccer_practice.jpg', 0), (0, 0), fx=0.8, fy=0.8)
+template = cv2.resize(cv2.imread('assets/shoe.PNG', 0), (0, 0), fx=0.8, fy=0.8)
+h, w = template.shape
+
+methods = [cv2.TM_CCOEFF, cv2.TM_CCOEFF_NORMED, cv2.TM_CCORR,
+            cv2.TM_CCORR_NORMED, cv2.TM_SQDIFF, cv2.TM_SQDIFF_NORMED]
+
+for method in methods:
+    img2 = img.copy()
+
+    result = cv2.matchTemplate(img2, template, method)
+    min_val, max_val, min_loc, max_loc = cv2.minMaxLoc(result)
+    if method in [cv2.TM_SQDIFF, cv2.TM_SQDIFF_NORMED]:
+        location = min_loc
+    else:
+        location = max_loc
+
+    bottom_right = (location[0] + w, location[1] + h)    
+    cv2.rectangle(img2, location, bottom_right, 255, 5)
+    cv2.imshow('Match', img2)
+    cv2.waitKey(0)
+    cv2.destroyAllWindows()