微信公众号搜"智元新知"关注
微信扫一扫可直接关注哦!

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

如何解决如何查看一个图像是否包含在另一个图像图像中,如果有则输出为 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):

版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。