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

利用torchaudio库中的save()函数将音频数据保存到指定目录

发布时间:2023-12-18 06:20:33

torchaudio库是一个基于PyTorch的音频处理库,提供了处理音频数据的各种功能和工具。其中,save()函数用于将音频数据保存到指定目录。

以下是一个使用例子,演示了如何使用save()函数将音频数据保存到指定目录:

import torch
import torchaudio

# 创建示例音频数据
waveform = torch.randn(2, 48000)  # 生成一个2通道的随机音频数据,采样率为48000Hz

# 保存音频数据
save_path = "path/to/save/audio.wav"
torchaudio.save(save_path, waveform, sample_rate=48000)

# 加载并验证保存的音频数据
loaded_waveform, sr = torchaudio.load(save_path)

print("原始音频数据形状:", waveform.shape)
print("加载的音频数据形状:", loaded_waveform.shape)
print("原始音频数据采样率:", 48000)
print("加载的音频数据采样率:", sr)

上述代码中,首先使用torch.randn()函数生成一个随机的音频数据waveform,其中waveform是一个2通道的音频数据,采样率为48000Hz。然后,使用torchaudio.save()函数将音频数据保存到指定目录save_path,使用sample_rate参数指定采样率为48000Hz。

接着,使用torchaudio.load()函数加载保存的音频数据,并将加载的音频数据赋值给loaded_waveform,将加载的采样率赋值给sr

最后,打印原始音频数据和加载的音频数据的形状以及原始音频数据和加载的音频数据的采样率。

通过上述例子,我们可以看到使用save()函数将音频数据保存到指定目录非常简单,只需要指定保存路径和音频数据即可。