通过不同的SNR修改噪声强度

Question

我正在尝试信号处理库ThinkDSP。

我想要的操作是在原始.wav文件上添加一些噪音

我知道add_noise = my_wav + noise可以很容易地完成此操作

但是，我想在不同的SNR条件下修改噪声

我如何通过ThinkDSP实现这一目标？或其他方法也欢迎。谢谢。

Answer 1

您需要计算wav文件中的功率并确定将为您提供所需snr的噪声功率。在带有scipy的Python中，这非常容易。

# Import scipy
import scipy as sp

# Signal power in data from wav file
psig = sig.var()

# For 10 dB SNR, calculate linear SNR (SNR = 10Log10(Psig/Pnoise)
snr_lin = 10.0**(10.0/10.0)

# Find required noise power
pnoise = psig/snr_lin

# Create noise vector
noise = sp.sqrt(pnoise)*sp.randn(len(sig))

# Add noise to signal
sig_plus_noise = sig + noise

有关更多信息，请参见此答案Add random noise with specific SNR to a signal