Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- import cv2
- import numpy as n
- import imutils
- img_rgb = cv2.imread('/Users/macbook/Downloads/e447a0c9-876b-431b-9c0f-abba6aa88c45.png')
- template = cv2.imread('/Users/macbook/Downloads/Waldo.png')
- w, h = template.shape[:-1]
- res = cv2.matchTemplate(img_rgb, template, cv2.TM_CCOEFF_NORMED)
- threshold = .8
- loc = n.where(res >= threshold)
- for pt in zip(*loc[::-1]): # Switch columns and rows
- cv2.rectangle(img_rgb, pt, (pt[0] + w, pt[1] + h), (0, 0, 255), 2)
- mask = n.zeros(img_rgb.shape, dtype = "uint8")
- img_rgb[cv2.rectangle(img_rgb, pt, (pt[0] + w, pt[1] + h), (0, 0, 255), 2)]
- #img_rgb = cv2.addWeighted(img_rgb, 0.3, mask, 0.7, 0)
- #cv2.imwrite("result.png"Iutils.resize(img_rgb, height = 650))
- cv2.imwrite('result.png', img_rgb)
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement