matlab提取幅值的STFT函数

以下是一个示例代码，用于在Matlab中提取幅值的STFT函数：

function [stft, f, t] = stft_amplitude(x, fs, win, hop, nfft)
% x: input signal
% fs: sampling frequency
% win: window function
% hop: hop size
% nfft: FFT size

% STFT
[S, f, t] = spectrogram(x, win, win-hop, nfft, fs);

% Amplitude
stft = abs(S);

end

该函数使用Matlab的内置`spectrogram`函数进行STFT计算，并返回幅值矩阵。输入参数包括：

- x：输入信号
- fs：采样频率
- win：窗函数
- hop：跳跃大小
- nfft：FFT大小

输出参数包括：

- stft：STFT幅值矩阵
- f：频率向量
- t：时间向量

可以使用以下代码来调用该函数：

% Load audio file
[x, fs] = audioread('example.wav');

% Parameters
win = hamming(1024);
hop = 512;
nfft = 2048;

% STFT amplitude
[stft, f, t] = stft_amplitude(x, fs, win, hop, nfft);

% Plot
imagesc(t, f, 20*log10(stft));
axis xy;
xlabel('Time (s)');
ylabel('Frequency (Hz)');
colorbar;

该代码将加载名为`example.wav`的音频文件，并绘制其STFT幅值。