在声音分类中常常需要 在声音中加入指定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)