torch_audiomentations_compose_gain_polarity_augmentation_batch.py

1import torch
2from torch_audiomentations import Compose, Gain, PolarityInversion
3
4# Initialize augmentation callable
5apply_augmentation = Compose(
6    transforms=[
7        Gain(
8            min_gain_in_db=-15.0,
9            max_gain_in_db=5.0,
10            p=0.5,
11        ),
12        PolarityInversion(p=0.5),
13    ]
14)
15
16# Create a dummy batch of 3 audio samples (1 channel, 32000 samples each)
17# Shape: (batch_size, num_channels, num_samples)
18dummy_audio_batch = torch.randn(3, 1, 32000)
19
20# Apply the augmentation to the batch
21# This can also be done on the GPU if the input tensor is on the GPU
22augmented_audio_batch = apply_augmentation(
23    dummy_audio_batch, sample_rate=16000
24)