Python實現屏幕錄制功能的代碼
前段時間做視頻時需要演示電腦端的操作,因此要用到屏幕錄制,下載了個迅捷屏幕錄制,但是沒有vip錄制的視頻有水印且只能錄制二分鐘,于是鄙人想了下能不能通過萬能的python來實現呢?經過一晚上的嘗試發現這條路是可以走的通的。分享一下自己的想法,整體思路是PIL模塊中的ImageGrab不停的獲得當前屏幕,利用opencv寫入視頻流話不多說,直接上代碼,有什么更好的建議,歡迎大家交流!
'''python + opencv 實現屏幕錄制_by-_Zjh_'''from PIL import ImageGrabimport numpy as npimport cv2p = 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’): breakvideo.release()cv2.destroyAllWindows()
在停止錄制的時候我自己采用的是opencv的q鍵盤退出,實際上還是會有個黑框彈出來,如果有更好的退出錄制方式請留下您的建議,本次還有個不足的地方是沒有增加錄制聲音的功能。最近事情太多了,等放假了出一個完整版的GUI程序。好了今天就先到這了,快要考線性系統這門課了,得開始復習了
最近接觸了 pynput 這個強大的三方庫,可以全局監聽鍵盤、鼠標事件。把以前的屏幕錄制的程序改進一下!請忽略以前的程序,直接看這個!
'''python 屏幕錄制改進版,無opencv黑框顯示!@zhou 2020/1/29_'''from PIL import ImageGrabimport numpy as npimport cv2import datetimefrom pynput import keyboardimport threadingflag=False #停止標志位def video_record(): ''' 屏幕錄制! :return: ''' name = datetime.datetime.now().strftime(’%Y-%m-%d %H-%M-%S’) #當前的時間 p = ImageGrab.grab() # 獲得當前屏幕 a, b = p.size # 獲得當前屏幕的大小 fourcc = cv2.VideoWriter_fourcc(*’XVID’) # 編碼格式 video = cv2.VideoWriter(’%s.avi’%name, fourcc, 20, (a, b)) # 輸出文件命名為test.mp4,幀率為16,可以自己設置 while True: im = ImageGrab.grab() imm=cv2.cvtColor(np.array(im), cv2.COLOR_RGB2BGR)#轉為opencv的BGR格式 video.write(imm) if flag: print('錄制結束!') break video.release()def on_press(key): ''' 鍵盤監聽事件!!! :param key: :return: ''' #print(key) global flag if key == keyboard.Key.esc: flag=True print('stop monitor!') return False #返回False,鍵盤監聽結束!if __name__==’__main__’: th=threading.Thread(target=video_record) th.start() with keyboard.Listener(on_press=on_press) as listener: listener.join()
現在錄制就不用之前那個黑框了,按 esc按鍵 直接停止錄制!
PS:在通過代碼給大家介紹下python 實現屏幕錄制
PIL 即pollow 的安裝命令如下:
pip install pillow
其中cv2的安裝是下面這條命令
pip install opencv-python
代碼實現:
# coding: utf-8from PIL import ImageGrabimport numpy as npimport cv2fps = 20start = 3 # 延時錄制end = 15 # 自動結束時間curScreen = ImageGrab.grab() # 獲取屏幕對象height, width = curScreen.sizevideo = cv2.VideoWriter(’video02.avi’, cv2.VideoWriter_fourcc(*’XVID’), fps, (height, width))imageNum = 0while True: imageNum += 1 captureImage = ImageGrab.grab() # 抓取屏幕 frame = cv2.cvtColor(np.array(captureImage), cv2.COLOR_RGB2BGR) # 顯示無圖像的窗口 cv2.imshow(’capturing’, np.zeros((1, 255), np.uint8)) # 控制窗口顯示位置,方便通過按鍵方式退出 cv2.moveWindow(’capturing’, height - 100, width - 100) if imageNum > fps * start: video.write(frame) # 退出條件 if cv2.waitKey(50) == ord(’q’) or imageNum > fps * end: breakvideo.release()cv2.destroyAllWindows()
總結
到此這篇關于Python實現屏幕錄制功能的文章就介紹到這了,更多相關python 屏幕錄制內容請搜索好吧啦網以前的文章或繼續瀏覽下面的相關文章希望大家以后多多支持好吧啦網!
相關文章:
