python 通过麦克风录音 生成wav文件的方法
程序员文章站
2024-02-05 23:03:34
如下所示：
#!/usr/bin/env python
# -*- coding: utf-8 -*-
##...
如下所示：
#!/usr/bin/env python
# -*- coding: utf-8 -*-
########################################################################
#
# copyright (c) 2017 aibot.me, inc. all rights reserved
#
########################################################################
"""
file: gen_wav.py
date: 2017/03/24 12:36:27
brief: 通过麦克风录音 生成 wav文件
"""
import os
import sys
import wave
import numpy as np
from datetime import datetime
from pyaudio import pyaudio, paint16
class genaudio(object):
def __init__(self):
self.num_samples = 2000 #pyaudio内置缓冲大小
self.sampling_rate = 8000 #取样频率
self.level = 1500 #声音保存的阈值
self.count_num = 20 #count_num个取样之内出现count_num个大于level的取样则记录声音
self.save_length = 8 #声音记录的最小长度：save_length * num_samples 个取样
self.time_count = 8 #录音时间，单位s
self.voice_string = []
#保存文件
def save_wav(self, filename):
wf = wave.open(filename, 'wb')
wf.setnchannels(1)
wf.setsampwidth(2)
wf.setframerate(self.sampling_rate)
wf.writeframes(np.array(self.voice_string).tostring())
wf.close()
def read_audio(self):
pa = pyaudio()
stream = pa.open(format=paint16, channels=1, rate=self.sampling_rate, input=true,
frames_per_buffer=self.num_samples)
save_count = 0
save_buffer = []
time_count = self.time_count
while true:
time_count -= 1
# 读入num_samples个取样
string_audio_data = stream.read(self.num_samples)
# 将读入的数据转换为数组
audio_data = np.fromstring(string_audio_data, dtype = np.short)
#计算大于 level 的取样的个数
large_sample_count = np.sum(audio_data > self.level)
print(np.max(audio_data)), "large_sample_count=>", large_sample_count
# 如果个数大于count_num，则至少保存save_length个块
if large_sample_count > self.count_num:
save_count = self.save_length
else:
save_count -= 1
if save_count < 0:
save_count = 0
if save_count > 0:
save_buffer.append(string_audio_data)
else:
if len(save_buffer) > 0:
self.voice_string = save_buffer
save_buffer = []
print("recode a piece of voice successfully!")
return true
if time_count == 0:
if len(save_buffer) > 0:
self.voice_string = save_buffer
save_buffer = []
print("recode a piece of voice successfully!")
return true
else:
return false
return true
if __name__ == "__main__":
r = genaudio()
r.read_audio()
r.save_wav("./test.wav")
以上这篇python 通过麦克风录音 生成wav文件的方法就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持。