欢迎访问宙启技术站
智能推送

使用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()函数可以很方便地保存音频文件到指定路径,以供后续处理或分析。