利用Python来完成屏幕录制

前段时间做视频时需要演示电脑端的操作，因此要用到屏幕录制，下载了个迅捷屏幕录制，但是没有vip录制的视频有水印且只能录制二分钟，于是鄙人想了下能不能通过万能的python来实现呢？经过一晚上的尝试发现这条路是可以走的通的。分享一下自己的想法，整体思路是PIL模块中的ImageGrab不停的获得当前屏幕，利用opencv写入视频流话不多说，直接上代码，有什么更好的建议，欢迎大家交流！

"""python + opencv 实现屏幕录制_by-_Zjh_"""
from PIL import ImageGrab
import numpy as np
import cv2
p = ImageGrab.grab()#获得当前屏幕
k=np.zeros((200,200),np.uint8)
a,b=p.size#获得当前屏幕的大小
fourcc = cv2.VideoWriter_fourcc(*'XVID')#编码格式
video = cv2.VideoWriter('test.avi', fourcc, 16, (a, b))#输出文件命名为test.mp4,帧率为16，可以自己设置
while True:
    im = ImageGrab.grab()
    imm=cv2.cvtColor(np.array(im), cv2.COLOR_RGB2BGR)#转为opencv的BGR格式
    video.write(imm)
    cv2.imshow('imm', k)
    if cv2.waitKey(1) & 0xFF == ord('q'):
        break
video.release()
cv2.destroyAllWindows()

在停止录制的时候我自己采用的是opencv的q键盘退出，实际上还是会有个黑框弹出来，如果有更好的退出录制方式请留下您的建议，本次还有个不足的地方是没有增加录制声音的功能。最近事情太多了，等放假了出一个完整版的GUI程序。好了今天就先到这了，快要考线性系统这门课了，得开始复习了