问题描述
我有一个对屏幕进行截图的脚本,我想看看里面是否有较小的图像。我目前正在使用 pyautogui 截取屏幕截图。我知道我可以使用 open-cv,但我不知道如何将其转换为布尔值,例如当检测到较小的图像时给出 1 或类似 True 以及不给出 0 或 False 时。谢谢!
解决方法
您可以使用 cv2.matchTemplate
方法:
import cv2
import numpy as np
img1 = cv2.imread("image1.png")
img2 = cv2.imread("image2.png")
img1_gray = cv2.cvtColor(img1,cv2.COLOR_BGR2GRAY)
img2_gray = cv2.cvtColor(img2,cv2.COLOR_BGR2GRAY)
res = cv2.matchTemplate(img1_gray,img2_gray,cv2.TM_CCOEFF_NORMED)
if np.any(res > 0.9):
print("Image2 is in Image1.")
是 0.9
表示将图像分类为其他图像内部所需的置信度。为了更准确,您可以执行以下操作:
if np.any(res > 0.99):