如何查看一个图像是否包含在另一个图像图像中，如果存在则输出 1，如果不存在则输出 0？

Question

我有一个截取屏幕截图的脚本，我想看看其中是否有较小的图像。我目前正在使用 pyAutoGui 来截取屏幕截图。我知道我可以使用 open-cv，但我不知道如何将其转换为布尔值，例如当检测到较小的图像时给出 1 或类似 True，以及何时不给出 0 或 False。谢谢！

Answer 1

您可以使用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):

如何查看一个图像是否包含在另一个图像图像中，如果存在则输出 1，如果不存在则输出 0？

How to see if an image is contained in another image image and get an output of 1 if there is or 0 if there isn't?

python

opencv

pyautogui