Spaces:
Building
Building
File size: 492 Bytes
f6b56a2 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 |
from gyraudio.audio_separation.data.dataset import AudioDataset
import logging
import torchaudio
class SingleAudioDataset(AudioDataset):
def load_data(self):
self.file_list = sorted(list(self.data_path.glob("*.wav")))
def __getitem__(self, idx: int):
audio_path = self.file_list[idx]
assert audio_path.exists()
audio_signal, sampling_rate = torchaudio.load(str(audio_path))
logging.debug(f"{audio_signal.shape}")
return audio_signal
|