generate_fft_for_all_files_in_directory.py 文件源码

def generate_and_save_fft(wave_file):
    sample_rate, X = sp.io.wavfile.read(wave_file)
    Y = abs((sp.fft(X))[:100000])   # change if different number of components are desired,
                                    # max value for genres dataset is more than 600000
    fft_file = wave_file[:-3] + "fft"
#    print fft_file
    np.save(fft_file, Y)

#test