使用Python 打造专属自己的屏幕录制工具：使用NumPy、PIL和OpenCV的完整指南

LIY若依

已于 2024-08-07 23:47:19 修改

阅读量1.3k

点赞数 18

文章标签： opencv 人工智能 python 应用开发录屏软件

于 2024-08-07 23:47:05 首次发布

本文链接：https://blog.youkuaiyun.com/m0_74972192/article/details/141003955

版权

简介

在这篇博客中，我们将介绍如何使用 NumPy、PIL 和 OpenCV 创建一个屏幕录制工具。通过逐步解析代码片段，解释每个部分的功能，最终展示完整代码。希望这篇博客能帮助你理解如何实现屏幕录制功能。

依赖库

在运行代码之前，请确保已安装以下依赖库：

NumPy
PIL（Pillow）
OpenCV

可以使用以下命令安装这些库：

pip install numpy pillow opencv-python

步骤 1: 导入必要的库

首先，我们需要导入必要的库，包括 NumPy、PIL 的 ImageGrab 模块和 OpenCV。

import numpy as np
from PIL import ImageGrab
import cv2

步骤 2: 设置录制参数

接下来，我们设置录制参数，包括屏幕尺寸、文件名和帧率。

# 设置录制参数
SCREEN_SIZE = (1920, 1080)
FILENAME = 'recorded_video.avi'
FPS = 30.0

解析

SCREEN_SIZE：设置屏幕尺寸为 1920x1080。
FILENAME：设置录制视频的文件名为 recorded_video.avi。
FPS：设置录制视频的帧率为 30 帧每秒。

步骤 3: 初始化视频写入对象

我们使用 OpenCV 初始化视频写入对象。

# 开始录制
fourcc = cv2.VideoWriter_fourcc(*"XVID")
out = cv2.VideoWriter(FILENAME, fourcc, FPS, SCREEN_SIZE)

解析

cv2.VideoWriter_fourcc(*"XVID")：设置视频编码格式为 XVID。
cv2.VideoWriter(FILENAME, fourcc, FPS, SCREEN_SIZE)：初始化视频写入对象，指定文件名、编码格式、帧率和屏幕尺寸。

步骤 4: 开始录制

我们使用一个循环来捕获屏幕截图并写入视频文件。

cnt = 0
while True:
    # 获取屏幕截图
    img = ImageGrab.grab(bbox=(0, 0, 1920, 1080))
    print('recordin..')
    # 转换为OpenCV格式
    frame = cv2.cvtColor(np.array(img), cv2.COLOR_RGB2BGR)
    # 写入视频文件
    out.write(frame)

    cnt = cnt + 1

    if cnt == 100:  #满100帧后终止循环
        break

解析

ImageGrab.grab(bbox=(0, 0, 1920, 1080))：捕获屏幕截图，指定捕获区域为 1920x1080。
cv2.cvtColor(np.array(img), cv2.COLOR_RGB2BGR)：将截图转换为 OpenCV 格式。
out.write(frame)：将转换后的帧写入视频文件。
cnt = cnt + 1：计数器增加。
if cnt == 100：当计数器达到 100 时，终止循环。

步骤 5: 停止录制

我们在循环结束后释放视频写入对象并关闭所有 OpenCV 窗口。

# 停止录制
out.release()
cv2.destroyAllWindows()

解析

out.release()：释放视频写入对象。
cv2.destroyAllWindows()：关闭所有 OpenCV 窗口。

完整代码

以下是完整代码：

import numpy as np
from PIL import ImageGrab
import cv2

# 设置录制参数
SCREEN_SIZE = (1920, 1080)
FILENAME = 'recorded_video.avi'
FPS = 30.0

# 开始录制
fourcc = cv2.VideoWriter_fourcc(*"XVID")
out = cv2.VideoWriter(FILENAME, fourcc, FPS, SCREEN_SIZE)

cnt = 0
while True:
    # 获取屏幕截图
    img = ImageGrab.grab(bbox=(0, 0, 1920, 1080))
    print('recordin..')
    # 转换为OpenCV格式
    frame = cv2.cvtColor(np.array(img), cv2.COLOR_RGB2BGR)
    # 写入视频文件
    out.write(frame)

    cnt = cnt + 1

    if cnt == 100:  #满100帧后终止循环
        break

# 停止录制
out.release()
cv2.destroyAllWindows()