利用python 在音频中加入指定SNR的噪声

在声音分类中常常需要 在声音中加入指定SNR的噪声， 本文提供增加指定SNR的加性高斯白噪声（Additive White Gaussian Noise (AWGN)）的代码。

代码如下：

def awgn(audio, snr):
    #在audio y中 添加噪声 噪声强度SNR为int
    audio_power = audio ** 2 
    audio_average_power = np.mean(audio_power)
    audio_average_db = 10 * np.log10(audio_average_power)
    noise_average_db = audio_average_db - snr
    noise_average_power = 10 ** (noise_average_db / 10)
    mean_noise = 0 
    noise = np.random.normal(mean_noise, np.sqrt(noise_average_power), len(audio))
    return audio + noise

 audio为clean audio, snr为指定的整数大小。 

如 clean audio: waveform 与 spectrogram为

 

SNR指定10时候： waveform 与spectrogram为

 

敢于spectrogram 计算可以参考：理解梅尔频谱（mel spectrogram）