如何查看一个图像是否包含在另一个图像图像中,如果有则输出为 1,如果没有则为 0?

问题描述

我有一个对屏幕进行截图的脚本,我想看看里面是否有较小的图像。我目前正在使用 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):