利用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)

猜你喜欢

转载自blog.csdn.net/bo17244504/article/details/124754703
今日推荐