利用Python的键盘监听实现自动截屏测试PyTorch 简单逻辑回回归

import pynput
import time
import os
from PIL import ImageGrab

# 定义回调函数，如果按下空格键则打印“按下空格键”
def on_press(key):
    if key == pynput.keyboard.Key.space:
        print('按下空格键')
        # 调用截屏函数
        capture_screen(directory="jump")

# 定义回调函数，如果按下esc键则停止监听
def on_release(key):
    if key == pynput.keyboard.Key.esc:
        # 停止监听
        return False

# 定义截屏函数，截屏并保存到指定目录
def capture_screen(directory):
    # 检查目录是否存在，不存在则创建目录
    if not os.path.exists(directory):
        os.makedirs(directory)
    # 获取当前时间戳作为图片名称
    now = int(time.time())
    # 截屏并保存图片
    ImageGrab.grab().save(f"{directory}/{now}.jpg")

# 循环监听键盘事件
while True:
    with pynput.keyboard.Listener(on_press=on_press, on_release=on_release) as listener:
        listener.join()
    # 如果没有按下空格，则每隔1秒保存一张截屏到nonoe目录
    capture_screen(directory="none")
    time.sleep(1)