deepextractor.generation.generate_spectrograms

Convert time-domain .npy arrays to STFT spectrograms (magnitude + phase).

Also provides a utility to concatenate chunked spectrogram files.

Usage:

deepextractor-specgen --input-dir data/pycbc_noise/time_domain/ --output-dir data/pycbc_noise/spectrogram_domain/

Module Contents

deepextractor.generation.generate_spectrograms.DEFAULT_N_FFT = 512

deepextractor.generation.generate_spectrograms.DEFAULT_WIN_LENGTH = 64

deepextractor.generation.generate_spectrograms.DEFAULT_HOP_LENGTH = 32

deepextractor.generation.generate_spectrograms.apply_stft_and_save(array_path, save_path, n_fft, hop_length, win_length, window, chunk_size=5000)

Apply STFT to a .npy array in chunks and save the result.

deepextractor.generation.generate_spectrograms.load_and_concatenate_chunks(data_dir, base_filename, total_chunks)

Load and concatenate chunked numpy arrays saved as {base}_chunk_{i}.npy.

deepextractor.generation.generate_spectrograms.main()