加载数据关键在于:torch.utils.data.Dataloader(可以加载数据集,并将数据集转为Tensor形式)
#安装包
!pip install torchaudio
import torch
import torchaudio
#访问数据集
yesno_data=torchaudio.datasets.YESNO('./',download=True)
#加载数据集
data_loader=torch.utils.data.DataLoader(yesno_data,batch_size=1,shuffle=True)
#迭代数据
for data in data_loader:
print(data)
print(f'waveform{data[0]},sample_rate{data[1]},labels{data[2]}')
break
#可视化数据
import matplotlib.pyplot as plt
print(data[0][0].numpy())
print(waveform.t())
d=waveform.t().numpy()
print(type(d))
plt.plot(waveform.t().numpy())
plt.show()