import sensor, image, time

# 加载转换后的 PGM 模板图像
template_1 = image.Image("/output_image_1.pgm")  # 数字1的模板
template_2 = image.Image("/output_image_2.pgm")  # 数字2的模板
template_3 = image.Image("/output_image_3.pgm")  # 数字3的模板
template_4 = image.Image("/output_image_4.pgm")  # 数字4的模板

# 相机初始化
sensor.reset()
sensor.set_pixformat(sensor.GRAYSCALE)  # 使用灰度图像
sensor.set_framesize(sensor.QVGA)  # 设置分辨率为 QVGA
sensor.skip_frames(time=2000)  # 等待相机准备

while True:
    img = sensor.snapshot()  # 获取图像

    # 在图像中查找模板
    match_1 = img.find_template(template_1, 0.7)  # 模板匹配，阈值为 0.7
    match_2 = img.find_template(template_2, 0.7)
    match_3 = img.find_template(template_3, 0.7)
    match_4 = img.find_template(template_4, 0.7)
    
    # 如果找到数字1，绘制矩形并标注
    if match_1:
        img.draw_rectangle(match_1, color=(255, 0, 0))  # 红色矩形
        img.draw_string(match_1[0], match_1[1] - 10, "1", color=(0, 255, 0))  # 显示数字1
    
    # 如果找到数字2，绘制矩形并标注
    if match_2:
        img.draw_rectangle(match_2, color=(255, 0, 0))  # 红色矩形
        img.draw_string(match_2[0], match_2[1] - 10, "2", color=(0, 255, 0))  # 显示数字2
    
    # 如果找到数字3，绘制矩形并标注
    if match_3:
        img.draw_rectangle(match_3, color=(255, 0, 0))  # 红色矩形
        img.draw_string(match_3[0], match_3[1] - 10, "3", color=(0, 255, 0))  # 显示数字3
    
    # 如果找到数字4，绘制矩形并标注
    if match_4:
        img.draw_rectangle(match_4, color=(255, 0, 0))  # 红色矩形
        img.draw_string(match_4[0], match_4[1] - 10, "4", color=(0, 255, 0))  # 显示数字4

    time.sleep(100)  # 延时