添加高斯白噪音
# 添加高斯白噪音def gen_gaussian_noise(signal,SNR): #signal: 原始信号，SNR: 添加噪声的信噪比，return: 生成的噪声noise=np.random.randn(*signal.shape) # *signal.shape 获取样本序列的尺寸noise=noise-np.mean(noise)signal_power=(1/signal.shape[0])*np.sum(np.power(signal,2))noise_variance=signal_power/np.power(10,(SNR/10))noise=(np.sqrt(noise_variance)/np.std(noise))*noisereturn noise
验证信噪比
def check_snr(signal,noise):signal_power=(1/signal.shape[0])*np.sum(np.power(signal,2))# 0.5722037noise_power=(1/noise.shape[0])*np.sum(np.power(noise,2)) # 0.90688SNR=10*np.log10(signal_power/noise_power)return SNR
给信号加噪声程序
Noise_data = []X_train=X_train.reshape(5900,2048).detach().numpy() # tensor转换为ndarrayfor i in range(X_train.shape[0]):noise = gen_gaussian_noise(X_train[i,:],-2)new_signal = X_train[i,:] + noiseNoise_data.append(new_signal)SNR= check_snr(X_train[i,:],noise)
