torch_pitch_shift_audio_tensor_by_semitones_with_fast_stft.py

1import torch
2import torchaudio
3from torch_pitch_shift import pitch_shift, get_fast_stft_setting
4
5# Load audio
6samples, sample_rate = torchaudio.load("input.wav")
7
8# Set up fast STFT parameters for the given sample rate
9up_to_half_pitch_shift = 12
10hop_length, n_fft = get_fast_stft_setting(sample_rate, up_to_half_pitch_shift)
11
12# Shift pitch by 2 semitones
13shifted_samples = pitch_shift(samples, 2, sample_rate, n_fft=n_fft, hop_length=hop_length)
14
15# Save the result
16torchaudio.save("output.wav", shifted_samples, sample_rate)