Python实现指定区域桌面变化监控并报警

pip install tkinter numpy opencv-python pyautogui pygame pillow

import tkinter as tk
from PIL import Image, ImageTk
start_x, start_y, end_x, end_y = None, None, None, None
def select_screen_region():
    global start_x, start_y, end_x, end_y
    def on_press(event):
        global start_x, start_y
        start_x, start_y = event.x, event.y
    def on_move(event):
        global end_x, end_y
        end_x, end_y = event.x, event.y
        canvas.coords(rect, start_x, start_y, end_x, end_y)
    def on_release(event):
        global end_x, end_y
        end_x, end_y = event.x, event.y
        canvas.coords(rect, start_x, start_y, end_x, end_y)
    root = tk.Tk()
    root.title("选择监控区域")
    root.attributes('-fullscreen', True)
    screenshot = pyautogui.screenshot()
    screenshot.save('desktop_screenshot.png', format='png')
    canvas = tk.Canvas(root, bg='white')
    canvas.pack(fill=tk.BOTH, expand=True)
    desktop_image = ImageTk.PhotoImage(file='desktop_screenshot.png')
    canvas.create_image(0, 0, anchor=tk.NW, image=desktop_image)
    rect = canvas.create_rectangle(0, 0, 0, 0, outline='red')
    canvas.bind("<ButtonPress-1>", on_press)
    canvas.bind("<B1-Motion>", on_move)
    canvas.bind("<ButtonRelease-1>", on_release)
    root.mainloop()

import cv2
import numpy as np
import threading
# 初始化pygame库
pygame.init()
pygame.mixer.init()
# 全局变量，加载MP3文件
def load_alarm_sound(file_path):
    try:
        pygame.mixer.music.load(file_path)
    except pygame.error:
        print(f"无法加载音频文件：{file_path}")
# 封装一个播放MP3警报的函数
def play_alarm():
    try:
        # 播放MP3文件
        pygame.mixer.music.play()
    except pygame.error:
        print("播放警报音频失败")
# 调用load_alarm_sound函数，加载警报音频
load_alarm_sound("警报声.mp3")
should_exit = False
is_alarm = False
prev_frame_image, current_frame_image = None, None
def monitor(data_source):
    global should_exit, is_alarm, prev_frame_image, current_frame_image
    if data_source == "screenshot":
        prev_frame = capture_screen()
    elif data_source == "camera":
        cap = cv2.VideoCapture(0)
        prev_frame = capture_video_frame(cap)
    while not should_exit:
        current_frame = None
        if data_source == "screenshot":
            current_frame = capture_screen()
        elif data_source == "camera":
            current_frame = capture_video_frame(cap)
        if current_frame is not None:
            prev_gray = cv2.cvtColor(prev_frame, cv2.COLOR_BGR2GRAY)
            current_gray = cv2.cvtColor(current_frame, cv2.COLOR_BGR2GRAY)
            similarity = ssim(prev_gray, current_gray)
            threshold = 0.95
            if similarity < threshold and not is_alarm:
                print("检测到较大变化，进行报警！")
                is_alarm = True
                play_alarm()
                # 保存变化前后的照片
                prev_frame_image = prev_frame.copy()
                current_frame_image = current_frame.copy()
                # 在5秒后重置is_alarm标志
                threading.Timer(5, reset_alarm_flag).start()
            else:
                print("没有变化")
        # 更新前一帧图像
        prev_frame = current_frame
        time.sleep(5)  # 等待5秒后再获取下一帧
    # 释放资源
    if data_source == "camera":
        cap.release()
    cv2.destroyAllWindows()
def start_recognition():
    global should_exit
    should_exit = False
    print("开始监控")
    th_monitor = threading.Thread(target=monitor, args=("screenshot",))
    th_monitor.daemon = True
    th_monitor.start()
def end_program():
    global should_exit
    should_exit = True
    print("结束程序")
    sys.exit(1)

import sys
import tkinter as tk
import time
import cv2
import numpy as np
import pyautogui
import pygame
from skimage.metrics import structural_similarity as ssim
from PIL import Image, ImageTk
# ... 在此添加上面的选择监控区域和监控区域变化的函数 ...
# 创建主窗口
root = tk.Tk()
root.title("区域监控和变化报警")
# 添加按钮
btn_start = tk.Button(root, text="开始监控", command=start_recognition)
btn_end = tk.Button(root, text="结束程序", command=end_program)
# 使用pack布局并设置side参数为LEFT实现横向排列
btn_start.pack(side=tk.LEFT, padx=10, pady=10)
btn_end.pack(side=tk.LEFT, padx=10, pady=10)
# 进入事件循环
root.mainloop()