使用Pygame和OpenCV记录窗口的内容为视频

python 复制代码
from PIL import ImageGrab
import numpy as np

    pg.init()

    gl_version = (3, 3) # GL Version number (Major, Minor)

    pg.display.gl_set_attribute(pg.GL_CONTEXT_MAJOR_VERSION, gl_version[0])

    pg.display.gl_set_attribute(pg.GL_CONTEXT_MINOR_VERSION, gl_version[1])

    pg.display.gl_set_attribute(pg.GL_CONTEXT_PROFILE_MASK, pg.GL_CONTEXT_PROFILE_CORE)

    display_size = (1440, 720)

    pg.display.set_mode(display_size, pg.OPENGL | pg.DOUBLEBUF | pg.RESIZABLE)

    screen_info = pg.display.Info()

    screen_width, screen_height = screen_info.current_w, screen_info.current_h

    window_x, window_y = (screen_width - display_size[0])/2, (screen_height - display_size[1])/2 #for recording screen

    record_screen = True#False#

    if record_screen:

        fourcc = cv2.VideoWriter_fourcc(*'XVID')

        window_out = cv2.VideoWriter('window_output.avi', fourcc, 30.0, display_size)


    if record_screen:

        if window_out is not None:
            image = ImageGrab.grab(bbox=(window_x, window_y, window_x+display_size[0],         window_y+display_size[1]))

            #image.save('screenshot.png')

            image = np.array(image)

            image_bgr = cv2.cvtColor(image, cv2.COLOR_RGB2BGR)

            window_out.write(image_bgr)


    if record_screen:

        window_out.release()

这段代码使用Pygame和OpenCV在屏幕上创建一个OpenGL窗口,并记录窗口的内容为视频。

首先,它初始化了Pygame库:

复制代码
pg.init() 

然后,设置OpenGL的版本和配置:

复制代码
gl_version = (3, 3) # 设置OpenGL版本号为3.3 pg.display.gl_set_attribute(pg.GL_CONTEXT_MAJOR_VERSION, gl_version[0]) pg.display.gl_set_attribute(pg.GL_CONTEXT_MINOR_VERSION, gl_version[1]) pg.display.gl_set_attribute(pg.GL_CONTEXT_PROFILE_MASK, pg.GL_CONTEXT_PROFILE_CORE) 

接下来,设置窗口的大小和一些Pygame的窗口属性:

复制代码
display_size = (1440, 720) # 设置窗口大小为1440x720 pg.display.set_mode(display_size, pg.OPENGL | pg.DOUBLEBUF | pg.RESIZABLE) # 设置窗口模式为OpenGL、双缓冲和可调整大小 

然后,获取屏幕信息,计算窗口的位置:

复制代码
screen_info = pg.display.Info() screen_width, screen_height = screen_info.current_w, screen_info.current_h window_x, window_y = (screen_width - display_size[0])/2, (screen_height - display_size[1])/2 # 计算窗口的位置 

接下来,如果设置了record_screenTrue,则创建一个视频写入器以记录窗口的内容:

复制代码
record_screen = True if record_screen: fourcc = cv2.VideoWriter_fourcc(*'XVID') # 使用XVID编解码器 window_out = cv2.VideoWriter('window_output.avi', fourcc, 30.0, display_size) # 创建视频写入器 

然后,如果需要记录屏幕,就获取窗口的截图,并将其转换为OpenCV可以处理的格式:

复制代码
if record_screen: if window_out is not None: image = ImageGrab.grab(bbox=(window_x, window_y, window_x+display_size[0], window_y+display_size[1])) # 获取窗口的截图 image = np.array(image) # 将截图转换为NumPy数组 image_bgr = cv2.cvtColor(image, cv2.COLOR_RGB2BGR) # 将RGB格式转换为BGR格式(OpenCV默认的颜色顺序) window_out.write(image_bgr) # 将截图写入视频 

最后,如果需要记录屏幕,则释放视频写入器:

复制代码
if record_screen: window_out.release() # 释放视频写入器 

这样,就完成了创建OpenGL窗口并记录窗口内容的整个过程。

相关推荐
尝试经历体验1 小时前
pycharm突然不能正常运行
python·深度学习·pycharm
大千AI助手1 小时前
灾难性遗忘:神经网络持续学习的核心挑战与解决方案
人工智能·深度学习·神经网络·大模型·llm·持续学习·灾难性遗忘
jay神1 小时前
基于Python的商品爬取与可视化系统
爬虫·python·数据分析·毕业设计·可视化系统
七元权2 小时前
论文阅读-SelectiveStereo
论文阅读·深度学习·双目深度估计·selectivestereo
浪浪山齐天大圣2 小时前
python数据可视化之Matplotlib(8)-Matplotlib样式系统深度解析:从入门到企业级应用
python·matplotlib·数据可视化
F_D_Z2 小时前
【PyTorch】单对象分割
人工智能·pytorch·python·深度学习·机器学习
编程自留地2 小时前
18.4 查看订单
python·django·商城
wanzhong23332 小时前
学习triton-第1课 向量加法
开发语言·python·高性能计算
浊酒南街2 小时前
Pytorch基础入门4
人工智能·pytorch·python
艾醒2 小时前
探索大语言模型(LLM):Ollama快速安装部署及使用(含Linux环境下离线安装)
人工智能·深度学习·算法