[python小工具]实现屏幕录制

用python实现屏幕录制

PIL 即pollow 的安装命令如下：

pip install pillow

其中cv2的安装是下面这条命令

pip install opencv-python

#python + opencv 实现屏幕录制
from PIL import ImageGrab
import numpy as np
import cv2

screen = ImageGrab.grab()#获得当前屏幕

length,width=screen.size#获得当前屏幕的大小
video_decode_style = cv2.VideoWriter_fourcc(*'XVID')#编码格式
video = cv2.VideoWriter('a.avi', video_decode_style, 32, (length, width))#输出文件命名为a.mp4,帧率为32，可以调节
while True:
    im = ImageGrab.grab()
    imm=cv2.cvtColor(np.array(im), cv2.COLOR_RGB2BGR)#转为opencv的BGR格式
    video.write(imm)

    if cv2.waitKey(1) & 0xFF == ord('q'):
        break
video.release()
cv2.destroyAllWindows()