Spaces:
Runtime error
Runtime error
File size: 784 Bytes
1160793 95d8ea8 1160793 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 |
import torch
from torch.utils.data import Dataset
from pathlib import Path
import torchaudio
import numpy as np
from torchaudio.transforms import Resample
class Minimal(Dataset):
def __init__(self, cfg):
self.wavs = ['p232_284.wav', 'p232_071.wav', 'p257_171.wav']
self.dataset_path = cfg['validation']['path']
self.target_rate = cfg['dataloader']['sample_rate']
self.resampler = Resample(orig_freq=cfg['validation']['sample_rate'],
new_freq=cfg['dataloader']['sample_rate'])
def __len__(self):
return len(self.wavs)
def __getitem__(self, idx):
wav, rate = torchaudio.load(Path(self.dataset_path) / self.wavs[idx])
wav = self.resampler(wav)
return wav, self.target_rate
|