如何搜索一组pdf，只有一个页面的图像片段

import fitz  # PyMuPDF
import cv2
import numpy as np

def extract_images_from_pdf(pdf_path):
    images = []
    document = fitz.open(pdf_path)
    for page_num in range(len(document)):
        page = document.load_page(page_num)
        image_list = page.get_images(full=True)
        for img_index, img in enumerate(image_list):
            xref = img[0]
            base_image = page.get_image(xref)
            image_bytes = base_image["image"]
            image = cv2.imdecode(np.frombuffer(image_bytes, np.uint8), cv2.IMREAD_COLOR)
            images.append((page_num, img_index, image))
    return images

def search_image(images, template_image_path):
    template = cv2.imread(template_image_path)
    template_height, template_width, _ = template.shape
    for page_num, img_index, image in images:
        result = cv2.matchTemplate(image, template, cv2.TM_CCOEFF_NORMED)
        min_val, max_val, min_loc, max_loc = cv2.minMaxLoc(result)
        if max_val > 0.8:  # 设置匹配阈值
            print(f"Found match at page {page_num}, image index {img_index}")

# 使用示例
pdf_path = 'example.pdf'
template_image_path = 'template.png'
images = extract_images_from_pdf(pdf_path)
search_image(images, template_image_path)