32 lines
1.0 KiB
Python
32 lines
1.0 KiB
Python
|
import os
|
|||
|
|
|
|||
|
|
import soundcard
|
|||
|
|
import soundfile
|
|||
|
|
|
|||
|
|
|
|||
|
|
class RecordAudio:
|
|||
|
|
def __init__(self, channels=1, sample_rate=16000):
|
|||
|
|
# 录音参数
|
|||
|
|
self.channels = channels
|
|||
|
|
self.sample_rate = sample_rate
|
|||
|
|
|
|||
|
|
# 获取麦克风
|
|||
|
|
self.default_mic = soundcard.default_microphone()
|
|||
|
|
|
|||
|
|
def record(self, record_seconds=3, save_path=None):
|
|||
|
|
"""录音
|
|||
|
|
|
|||
|
|
:param record_seconds: 录音时间，默认3秒
|
|||
|
|
:param save_path: 录音保存的路径，后缀名为wav
|
|||
|
|
:return: 音频的numpy数据
|
|||
|
|
"""
|
|||
|
|
print("开始录音......")
|
|||
|
|
num_frames = int(record_seconds * self.sample_rate)
|
|||
|
|
data = self.default_mic.record(samplerate=self.sample_rate, numframes=num_frames, channels=self.channels)
|
|||
|
|
audio_data = data.squeeze()
|
|||
|
|
print("录音已结束!")
|
|||
|
|
if save_path is not None:
|
|||
|
|
os.makedirs(os.path.dirname(save_path), exist_ok=True)
|
|||
|
|
soundfile.write(save_path, data=data, samplerate=self.sample_rate)
|
|||
|
|
return audio_data
|