asteroid_filterbanks_stft_encoder_decoder_quickstart.py

1import torch
2from asteroid_filterbanks import STFTFB, Encoder, Decoder
3
4# Create the filterbank
5fb = STFTFB(n_filters=512, kernel_size=256, stride=128)
6
7# Create encoder and decoder
8enc = Encoder(fb)
9dec = Decoder(fb)
10
11# Create a dummy signal
12signal = torch.randn(1, 1, 16000)
13
14# Transform to time-frequency domain
15tf_rep = enc(signal)
16
17# Transform back to time domain
18out = dec(tf_rep)
19
20# Check the shape of the output
21print(f"Input shape: {signal.shape}")
22print(f"TF representation shape: {tf_rep.shape}")
23print(f"Output shape: {out.shape}")