matlab语音增强维纳滤波代码

以下是一份matlab语音增强维纳滤波的代码示例：

% 读取音频文件
[x, fs] = audioread('audio_file.wav');

% 设置参数
win = 256; % 窗口大小
hop = win/2; % 帧移
nfft = win; % FFT点数
noise = x(1:fs*5); % 前5秒为噪声
alpha = 0.99; % 平滑系数
beta = 2; % 噪声功率系数
mu = 0.98; % 调整参数

% 计算噪声功率谱
nfft2 = nfft/2+1;
noise_stft = spectrogram(noise, win, win-hop, nfft, fs);
noise_power = mean(abs(noise_stft).^2, 2);

% 初始化变量
x_len = length(x);
x_enhanced = zeros(x_len, 1);
x_old = zeros(win, 1);
G = ones(nfft2, 1);

% 处理每一帧
for i = 1:hop:x_len-win
    % 分帧
    x_frame = x(i:i+win-1);
    
    % STFT
    x_stft = fft(x_frame, nfft);
    x_mag = abs(x_stft(1:nfft2));
    x_phase = angle(x_stft(1:nfft2));
    
    % 计算信噪比
    noise_mag = sqrt(noise_power);
    snr = max(x_mag./noise_mag, 1e-6);
    
    % 计算增益函数
    G = alpha*G + (1-alpha)*max(snr-1, 0).^beta;
    
    % 计算估计信号
    x_hat_mag = G.*x_mag;
    x_hat_stft = x_hat_mag.*exp(1j*x_phase);
    x_hat = real(ifft([x_hat_stft; conj(x_hat_stft(end-1:-1:2))], nfft));
    
    % 重叠相加
    x_enhanced(i:i+win-1) = x_enhanced(i:i+win-1) + x_hat.*hanning(win);
end

% 播放增强后的音频
sound(x_enhanced, fs);

% 绘制波形图
t = (0:x_len-1)/fs;
figure;
plot(t, x, 'b', t, x_enhanced, 'r');
xlabel('Time (s)');
ylabel('Amplitude');
legend('Original', 'Enhanced');