使用torchaudio的save()函数保存音频文件到指定路径
发布时间:2023-12-18 06:18:16
torch.audio的save()函数用于将音频文件保存到指定的路径。它接受以下参数:
- filepath:要保存音频文件的路径,可以是本地文件系统的绝对路径或相对路径。
- tensor:待保存为音频文件的图像张量。
- sample_rate:音频文件的采样率,默认为16000Hz。
下面我们将使用save()函数的一个例子来保存一个音频文件。
首先,我们需要导入必要的库:
import torch import torchaudio
接下来,我们需要加载音频文件,并将其转换为张量:
waveform, sample_rate = torchaudio.load('input.wav')
然后,我们可以使用save()函数将张量保存为音频文件:
torchaudio.save('output.wav', waveform, sample_rate)
在上面的例子中,我们将名为'input.wav'的音频文件加载为张量,并将其保存为'output.wav'。默认情况下,保存的音频文件的采样率与输入文件的采样率相同。如果想要指定不同的采样率,可以通过sample_rate参数进行修改。
完整的代码如下:
import torch
import torchaudio
waveform, sample_rate = torchaudio.load('input.wav')
torchaudio.save('output.wav', waveform, sample_rate)
上述代码将加载名为'input.wav'的音频文件,并将其保存为'output.wav'。
使用save()函数可以很方便地保存音频文件到指定路径,以供后续处理或分析。
