计算图像在屏幕上出现的次数

import cv2
import numpy as np

# 加载图像模板
template = cv2.imread('template.png', 0)

# 读取视频流或图像
cap = cv2.VideoCapture('video.mp4')

# 获取模板的宽度和高度
w, h = template.shape[::-1]

# 初始化计数器
count = 0

while True:
    ret, frame = cap.read()
    if not ret:
        break

    # 转换为灰度图像
    gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)

    # 进行模板匹配
    res = cv2.matchTemplate(gray, template, cv2.TM_CCOEFF_NORMED)
    threshold = 0.8
    loc = np.where(res >= threshold)

    # 计数匹配到的图像
    for pt in zip(*loc[::-1]):
        cv2.rectangle(frame, pt, (pt[0] + w, pt[1] + h), (0, 255, 0), 2)
        count += 1

    # 显示结果
    cv2.imshow('Frame', frame)
    if cv2.waitKey(1) & 0xFF == ord('q'):
        break

cap.release()
cv2.destroyAllWindows()

print(f'图像出现次数: {count}')