# -*- coding: utf-8 -*-
"""
Created on Thu Nov 15 22:37:02 2018
#QQ群：476842922（歡迎加群討論學習）
@author: Administrator
"""
"""python + 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()