From 858e928378ab9a9fb088e0e271d19b885dab0a81 Mon Sep 17 00:00:00 2001 From: Francisco Aranda Date: Mon, 12 Jun 2017 21:51:41 +0200 Subject: [PATCH] fix (audio import): enable multiple encoding load --- data/data_loader.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/data/data_loader.py b/data/data_loader.py index f7b55efd..09e335ce 100644 --- a/data/data_loader.py +++ b/data/data_loader.py @@ -12,9 +12,14 @@ windows = {'hamming': scipy.signal.hamming, 'hann': scipy.signal.hann, 'blackman': scipy.signal.blackman, 'bartlett': scipy.signal.bartlett} +def __load_audio_file_safetly(path): + try: + return torchaudio.load(path.encode('utf-8')) # py3 fix + except: + return torchaudio.load(path) def load_audio(path): - sound, _ = torchaudio.load(path.encode('utf-8')) # py3 fix + sound, _ = __load_audio_file_safetly(path) sound = sound.numpy() if len(sound.shape) > 1: if sound.shape[1] == 1: