opencv實現錄屏

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'):#q鍵推出
        break
video.release()
cv2.destroyAllWindows()