用Python进行屏幕录制的实现

更新时间：2024年04月18日 10:14:04 作者：鹅毛在路上了

关于屏幕录制这个功能需求,之前用过基于ffmpeg的Capture录屏软件,但是fps拉高以后会变得很卡,声音也同样出现卡顿,所以本文给大家介绍了用Python进行屏幕录制的实现,感兴趣的朋友可以参考下

引言

关于屏幕录制这个功能需求，之前用过基于ffmpeg的Capture录屏软件，但是fps拉高以后会变得很卡，声音也同样出现卡顿。也自己尝试过在python中调用ffmpeg的库函数，效果也不尽人意。网络上下载了几款录屏软件，不是要收费就是下载到捆绑软件或广告很是心累，因此想借此机会重新研究一下屏幕录制软件的工作原理，同时当作一个小项目练手。

gpt3.5给出的代码

当我还在犹豫是否要精心翻阅一下ffmpeg的说明书时，没想到gpt已果断给出了参考答案，可以使用Python的PyAutoGUI库来实现录屏功能。以下是一个简单的示例代码：

import pyautogui
import cv2
import numpy as np

# 获取屏幕分辨率
screen_size = (1920, 1080)

# 设置视频编码器
fourcc = cv2.VideoWriter_fourcc(*"XVID")

# 创建视频写入对象
out = cv2.VideoWriter("output.avi", fourcc, 20.0, screen_size)

# 开始录屏
while True:
    # 获取屏幕截图
    img = pyautogui.screenshot()

    # 将截图转换为OpenCV格式
    frame = np.array(img)
    frame = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)

    # 写入视频
    out.write(frame)

    # 按下q键退出录屏
    if cv2.waitKey(1) == ord("q"):
        break

# 释放资源
out.release()
cv2.destroyAllWindows()

从给出的代码看来，屏幕录制的工作原理就是以一定的帧率不断地截图，然后将截取的图片以相同的帧率合成一个视频。

放在安装了pyautogui的python环境里运行后，果然不出意外的话就要出意外了。gpt3.5给出的代码卡在运行中的界面纹丝不动，按’q’回车也并未退出循环，再问gpt也死活给不出修改方案。

查阅资料后发现要先imshow()之后，在ui窗口区域内按键才能有效终止循环：

import numpy as np
import pyautogui
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)

while True:
    # 获取屏幕截图
    img = pyautogui.screenshot()
    # 转换为OpenCV格式
    frame = cv2.cvtColor(np.array(img), cv2.COLOR_RGB2BGR)
    # 写入视频文件
    out.write(frame)
    cv2.imshow('Frame', frame)
    cv2.resizeWindow('Frame', 1920, 1080)

    # 检测按键
    if cv2.waitKey(1) == ord('q'):
        break

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

程序是能运行了，但是效果依旧不好，窗口一直有递归的效果，而且导出的视频其实是无法播放的。

在这里插入图片描述

更换截图函数——ImageGrab.grab

pyautogui虽然能实现截图，并在imshow里展示出来，但是导出的视频却无法播放，考虑肯能涉及到具体视频编解码参数问题，有懂的朋友请在评论区分享。这里采用更换PIL库中的截图函数ImageGrab.grab，可以实现截图并导出视频了，接下来最大的问题就是解决递归现象。

import numpy as np
import pyautogui
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)

while True:
    # 获取屏幕截图
    img = pyautogui.screenshot()
    # 转换为OpenCV格式
    frame = cv2.cvtColor(np.array(img), cv2.COLOR_RGB2BGR)
    # 写入视频文件
    out.write(frame)
    cv2.imshow('Frame', frame)
    cv2.resizeWindow('Frame', 1920, 1080)

    # 检测按键
    if cv2.waitKey(1) == ord('q'):
        break

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

在这里插入图片描述

禁用imshow解决递归现象

视频处理时的递归现象其实非常常见，除了物理中的镜面效应（观察两个平行放置的镜子会出现递归的现象），

将摄像头对准显示器，显示器上的画面也会观察到递归的现象：

在这里插入图片描述

经尝试，将imshow()禁用后，改为帧计数的方式自定义终止循环就不会出现递归的问题了：

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 = pyautogui.screenshot()
    img = ImageGrab.grab(bbox=(0, 0, 1920, 1080))
    print('recordin..')
    # 转换为OpenCV格式
    frame = cv2.cvtColor(np.array(img), cv2.COLOR_RGB2BGR)
    # 写入视频文件
    out.write(frame)
    # cv2.imshow('Frame', frame)
    # cv2.resizeWindow('Frame', 1920, 1080)

    # # 检测按键
    # if cv2.waitKey(1) == ord('q'):
    #     break

    cnt = cnt + 1

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

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

在这里插入图片描述

通过修改img = ImageGrab.grab(bbox=(0, 0, 2560, 1600))中的参数可以自定义录屏区域，x,y,w,h分别代表左上角坐标（起始坐标）和图片宽度、高度。比如我的屏幕分辨率是2560*1600，那么设置为0, 0, 2560, 1600就是录制全屏：

在这里插入图片描述

这样，我们就可以基本实现用Python进行屏幕录制的功能了。动态图预览看上去分辨率不高是因为用的格式工厂把录制的视频转了gif，压缩前录制的视频其实蛮清楚的。

通过修改fps的值，我们还可以自行录制一些高刷新率的电影、游戏画面，fps越高，画面越流畅哦。

摄像头录制代码

类似的，也可以用python实现相机录像的功能：

import cv2
import cv2 as cv

# 打开摄像头
cap = cv2.VideoCapture(0)

fourcc = cv.VideoWriter_fourcc(*'XVID')
file_name = 'output'
output = cv.VideoWriter((file_name + '.avi'), fourcc, 24.0, (640, 480))  #设置文件名，fps，分辨率

while cap.isOpened():

    res, frame = cap.read()
    if not res:
        print("Frame Cannot Be Received")
        break

    # Flipping the frame horizontally to get correct orientation
    frame = cv2.flip(frame, 90)

    # Displaying the current frame
    output.write(frame)
    cv2.imshow('Frame', frame)

    # If no input is received for 1ms, or if the key 'x' is pressed, interpreter goes outside of the loop
    if cv2.waitKey(1) == ord('x'):
        break

# Releasing everything after coming out of loop
cap.release()
output.release()
cv2.destroyAllWindows()

后期需求

现在屏幕录制的问题基本解决了，要想做一个实用的屏幕录制软件，还需要加上音频录制，并设计一个便捷的UI界面。

小结

到此这篇关于用Python进行屏幕录制的实现的文章就介绍到这了,更多相关Python屏幕录制内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家！

您可能感兴趣的文章:

Python开发.exe小工具的详细步骤
这篇文章主要介绍了Python开发.exe小工具的详细步骤,本文给大家介绍的非常详细，对大家的学习或工作具有一定的参考借鉴价值，需要的朋友可以参考下
2021-01-01
python文件名和文件路径操作实例
下面小编就为大家带来一篇python文件名和文件路径操作实例。小编觉得挺不错的，现在就分享给大家，也给大家做个参考。一起跟随小编过来看看吧
2017-09-09
如何使用python爬虫爬取要登陆的网站
这篇文章主要介绍了如何使用python爬虫爬取要登陆的网站,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下
2019-07-07
Python导入父文件夹中模块并读取当前文件夹内的资源
这篇文章主要给大家介绍了关于Python导入父文件夹中模块并读取当前文件夹内资源的相关资料，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧
2020-11-11
Python中 pickle 模块的 dump() 和 load()&
Python 的 pickle 模块用于实现二进制序列化和反序列化，一个对象可以被序列化到文件中，然后可以从文件中恢复，这篇文章主要介绍了Python中 pickle 模块的 dump() 和 load() 方法详解,需要的朋友可以参考下
2022-12-12
Python实现输出程序执行进度百分比的方法
这篇文章主要介绍了Python实现输出程序执行进度百分比的方法,涉及Python数值运算与系统输出相关操作技巧,需要的朋友可以参考下
2017-09-09
python3+telnetlib实现简单自动测试示例详解
telnetlib 模块提供一个实现Telnet协议的类 Telnet，本文重点给大家介绍python3+telnetlib实现简单自动测试示例详解,需要的朋友可以参考下
2021-08-08
Python Pyecharts绘制桑基图分析用户行为路径
这篇文章主要为大家介绍了Python Pyecharts绘制桑基图分析用户行为路径，有需要的朋友可以借鉴参考下，希望能够有所帮助，祝大家多多进步，早日升职加薪
2022-05-05
解决django框架model中外键不落实到数据库问题
这篇文章主要介绍了解决django框架model中外键不落实到数据库问题，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
2020-05-05
安装pyhttpx解决ImportError: DLL load failed错误
这篇文章主要为大家介绍了安装pyhttpx解决ImportError: DLL load failed错误,有需要的朋友可以借鉴参考下,希望能够有所帮助,祝大家多多进步,早日升职加薪
2023-08-08