原文鏈接：https://blog.csdn.net/zhaoyun_zzz/article/details/84341801
Python+opencv+pyaudio實現帶聲音屏幕錄制
原創luke-skyworker 發布於2018-11-28 23:05:23 閱讀數 3463 收藏
展開
文章目錄
聲音錄制
視頻錄制（無聲音）
錄制的音頻與視頻合成為帶聲音的視頻
基於個人的愛好和現實的需求，決定用Python做一個屏幕錄制的腳本。因為要看一些加密的視頻，每次都要登錄，特別麻煩，遂決定用自己寫的腳本，將加密視頻的播放過程全程錄制下來，這樣以后看自己的錄播就好了。結合近期自己學習的內容，正好用Python來練練手，鞏固自己的學習效果。
經過多番搜索，決定采用Python+opencv+pyaudio來實現屏幕錄制。網上搜索到的錄屏，基本都是不帶聲音的，而我要實現的是帶聲音的屏幕錄制。下面就開始一步一步的實現吧。
聲音錄制
import pyaudio
import wave
import sys
CHUNK = 1024
if len(sys.argv) < 2:
print("Plays a wave file.\n\nUsage: %s filename.wav" % sys.argv[0])
sys.exit(-1)
wf = wave.open(sys.argv[1], 'rb')
p = pyaudio.PyAudio()
stream = p.open(format=p.get_format_from_width(wf.getsampwidth()),
channels=wf.getnchannels(),
rate=wf.getframerate(),
output=True)
data = wf.readframes(CHUNK)
while data != '':
stream.write(data)
data = wf.readframes(CHUNK)
stream.stop_stream()
stream.close()
p.terminate()
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
簡潔回調函數版音頻錄制
import pyaudio
import wave
import time
import sys
CHUNK = 1024
FORMAT = pyaudio.paInt16
CHANNELS = 2
RATE = 44100
RECORD_SECONDS = 10
WAVE_OUTPUT_FILENAME = "output.wav"
p = pyaudio.PyAudio()
wf = wave.open(WAVE_OUTPUT_FILENAME, 'wb')
wf.setnchannels(CHANNELS)
wf.setsampwidth(p.get_sample_size(FORMAT))
wf.setframerate(RATE)
time_count = 0
def callback(in_data, frame_count, time_info, status):
wf.writeframes(in_data)
if(time_count < 10):
return (in_data, pyaudio.paContinue)
else:
return (in_data, pyaudio.paComplete)
stream = p.open(format=p.get_format_from_width(wf.getsampwidth()),
channels=wf.getnchannels(),
rate=wf.getframerate(),
input=True,
stream_callback=callback)
stream.start_stream()
print("* recording")
while stream.is_active():
time.sleep(1)
time_count += 1
stream.stop_stream()
stream.close()
wf.close()
p.terminate()
print("* recording done!")
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
視頻錄制（無聲音）
from PIL import ImageGrab
import numpy as np
import cv2
image = ImageGrab.grab()#獲得當前屏幕
width = image.size[0]
height = image.size[1]
print("width:", width, "height:", height)
print("image mode:",image.mode)
k=np.zeros((width,height),np.uint8)
fourcc = cv2.VideoWriter_fourcc(*'XVID')#編碼格式
video = cv2.VideoWriter('test.avi', fourcc, 25, (width, height))
#輸出文件命名為test.mp4,幀率為16，可以自己設置
while True:
img_rgb = ImageGrab.grab()
img_bgr=cv2.cvtColor(np.array(img_rgb), cv2.COLOR_RGB2BGR)#轉為opencv的BGR格式
video.write(img_bgr)
cv2.imshow('imm', img_bgr)
if cv2.waitKey(1) & 0xFF == ord('q'):
break
video.release()
cv2.destroyAllWindows()
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
錄制的音頻與視頻合成為帶聲音的視頻
錄制200幀，帶音頻的MP4視頻，單線程
import wave
from pyaudio import PyAudio,paInt16
from PIL import ImageGrab
import numpy as np
import cv2
from moviepy.editor import *
from moviepy.audio.fx import all
import time
CHUNK = 1024
FORMAT = pyaudio.paInt16
CHANNELS = 2
RATE = 44100
WAVE_OUTPUT_FILENAME = "output.wav"
p = pyaudio.PyAudio()
wf = wave.open(WAVE_OUTPUT_FILENAME, 'wb')
wf.setnchannels(CHANNELS)
wf.setsampwidth(p.get_sample_size(FORMAT))
wf.setframerate(RATE)
audio_record_flag = True
def callback(in_data, frame_count, time_info, status):
wf.writeframes(in_data)
if audio_record_flag:
return (in_data, pyaudio.paContinue)
else:
return (in_data, pyaudio.paComplete)
stream = p.open(format=p.get_format_from_width(wf.getsampwidth()),
channels=wf.getnchannels(),
rate=wf.getframerate(),
input=True,
stream_callback=callback)
image = ImageGrab.grab()#獲得當前屏幕
width = image.size[0]
height = image.size[1]
print("width:", width, "height:", height)
print("image mode:",image.mode)
k=np.zeros((width,height),np.uint8)
fourcc = cv2.VideoWriter_fourcc(*'XVID')#編碼格式
video = cv2.VideoWriter('test.mp4', fourcc, 9.5, (width, height))
#經實際測試，單線程下最高幀率為10幀/秒，且會變動，因此選擇9.5幀/秒
#若設置幀率與實際幀率不一致，會導致視頻時間與音頻時間不一致
print("video recording!!!!!")
stream.start_stream()
print("audio recording!!!!!")
record_count = 0
while True:
img_rgb = ImageGrab.grab()
img_bgr=cv2.cvtColor(np.array(img_rgb), cv2.COLOR_RGB2BGR)#轉為opencv的BGR格式
video.write(img_bgr)
record_count += 1
if(record_count > 200):
break
print(record_count, time.time())
audio_record_flag = False
while stream.is_active():
time.sleep(1)
stream.stop_stream()
stream.close()
wf.close()
p.terminate()
print("audio recording done!!!!!")
video.release()
cv2.destroyAllWindows()
print("video recording done!!!!!")
print("video audio merge!!!!!")
audioclip = AudioFileClip("output.wav")
videoclip = VideoFileClip("test.mp4")
videoclip2 = videoclip.set_audio(audioclip)
video = CompositeVideoClip([videoclip2])
video.write_videofile("test2.mp4",codec='mpeg4')
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
看來要提高幀率必須使用隊列加多線程了，這一步等到以后來添加吧。不過總是覺得用OpenCV來實現視頻錄制，有點怪異，畢竟opencv是用來做圖像與視頻分析的，還是走正道認真搗鼓opencv該做的事情吧。
————————————————
版權聲明：本文為CSDN博主「luke-skyworker」的原創文章，遵循 CC 4.0 BY-SA 版權協議，轉載請附上原文出處鏈接及本聲明。