spectrogram.py 文件源码

def create_spectrogram(wav):
    spec = stft.stft(wav, fftsize=fft_size)

    num_samples = spec.shape[0]

    # zero DC component
    #spec[0:num_samples,0:1] = 0.0

    # zero part where voice never is
    #spec[0:num_samples,0:10] = 0.0

    spec *= spec_norm

    spec_mags = np.sqrt(np.square(scipy.real(spec)) + np.square(scipy.imag(spec)))

    return spec, spec_mags