用python實現錄屏,親測好用!!!

前言

由與上不了學,教我們Mastercam的老師提前給我們佈置了4道習題。對我們太好了,謝謝老師😭
在這裏插入圖片描述
沒辦法,幹就完了。

只是要求附上製作過程視頻,就想到了能不能用python做個錄屏的程序,於是在網上看了好多文章,發現很多都不是很好用,要不就是太麻煩(我就錄個屏,不用聲音,也不用控件)。最後看到了一個符合我要求的,可以去看看這裏(感謝!)

不過呢,俗話說:適合自己的纔是最好的。當我們用的時候,會發現錄製的時間和視頻播放的時間不一致,有的快了,有的慢了,所以就想出了原因與解決辦法。接下來先上代碼吧

需要安裝的第三方庫有pillow、opencv-python、numpy、pynput

上‘馬’

import time,threading
from datetime import datetime
from PIL import ImageGrab
from cv2 import *
import numpy as np
from pynput import keyboard



def video_record():     # 錄入視頻
    global name
    name = datetime.now().strftime('%Y-%m-%d %H-%M-%S')  # 當前的時間(當文件名)
    screen = ImageGrab.grab()  # 獲取當前屏幕
    width, high = screen.size  # 獲取當前屏幕的大小
    fourcc = VideoWriter_fourcc('X', 'V', 'I', 'D')  # MPEG-4編碼,文件後綴可爲.avi .asf .mov等
    video = VideoWriter('%s.avi' % name, fourcc, 15, (width, high))  # (文件名,編碼器,幀率,視頻寬高)
    #print('3秒後開始錄製----')   # 可選
    #time.sleep(3)
    print('開始錄製!')
    global start_time
    start_time = time.time()
    while True:
        if flag:
            print("錄製結束!")
            global final_time
            final_time = time.time()
            video.release()  #釋放
            break
        im = ImageGrab.grab()   # 圖片爲RGB模式
        imm = cvtColor(np.array(im), COLOR_RGB2BGR)  # 轉爲opencv的BGR模式
        video.write(imm)   #寫入
        # time.sleep(5) # 等待5秒再次循環


def on_press(key):      # 監聽按鍵
    global flag
    if key == keyboard.Key.home:
        flag = True  # 改變
        return False  # 返回False,鍵盤監聽結束!


def video_info():      # 視頻信息
    video = VideoCapture('%s.avi' % name)   # 記得文件名加格式不要錯!
    fps = video.get(CAP_PROP_FPS)
    Count = video.get(CAP_PROP_FRAME_COUNT)
    size = (int(video.get(CAP_PROP_FRAME_WIDTH)), int(video.get(CAP_PROP_FRAME_HEIGHT)))
    print('幀率=%.1f'%fps)
    print('幀數=%.1f'%Count)
    print('分辨率',size)
    print('視頻時間=%.3f秒'%(int(Count)/fps))
    print('錄製時間=%.3f秒'%(final_time-start_time))
    print('推薦幀率=%.2f'%(fps*((int(Count)/fps)/(final_time-start_time))))


if __name__ == '__main__':
    flag = False
    th = threading.Thread(target=video_record)
    th.start()
    with keyboard.Listener(on_press=on_press) as listener:
        listener.join()
    time.sleep(1)   # 等待視頻釋放過後
    video_info()

解決

觀察代碼可以發現實現錄屏的方式就是不斷循環獲取屏幕然後寫入,可以想到每循環一次就是一幀,而循環一次要多長時間就取決於運行的速度了。所以我們只要讓一秒中寫入的幀數(也就是1秒中循環的次數)和video中的參數幀率相等,就能獲得差不多正常的速度

因此我在代碼中加入了視頻信息,算出了視頻播放的時間和錄製時間,再推算出來正常速度的幀率👉推薦幀率。只要先運行試一下,看一下推薦幀率是多少,再把代碼中的參數改過來就行了(要取整數哦)
在這裏插入圖片描述

要按home鍵來正常結束程序,纔會有video_info出現

然後呢!我發現老師要我們附製造過程視頻,我TM也不可能發個幾個幾百M甚至上G的視頻過去吧,老師也沒時間去看吧

於是乎,我在循環中加入了‘等待5秒’,這樣就差不多每5秒獲取1幀,推薦幀率顯示0.20,當然,我不能也不會按這個速度播放,給它來個25倍(幀率爲0.20*25=5,這樣視頻出來就又小又短了🤣)

多提一下,因爲兩個線程並不會同時結束,只是通過flag這個變量改變來使得另一個結束,所以按home鍵flag改變的時候,另一個線程video_record還處於運行循環中,因此我們要等它經過這次循環,進入下一次條件判斷,視頻才得以從內存釋放…呃嗯,就是我要在代碼倒數第2行那裏等待個5+1=6秒(這樣比較保險),否則的話,那個幀數會顯示爲0,甚至會出現運行錯誤(不能除0)

說了那麼多,怎麼能不給你們看一下效果呢

成果

4題,錄了4個,最小的不到4M,最大也就14M多點👏在這裏插入圖片描述
因爲這裏放不了AVI,所以我把最小的那個轉成GIF格式放給你們看看。
不過模糊了好多,還多了一些莫名的××點點😒
在這裏插入圖片描述

最後
在這裏插入圖片描述
😁😁😁😁😁😁😁😁😁😁😁😁😁😁😁😁😁

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章