From 23f50f828d7976e600782c58efc5cbdd1f761e68 Mon Sep 17 00:00:00 2001 From: Kais Kermani <47240396+KaisKermani@users.noreply.github.com> Date: Mon, 27 Mar 2023 22:38:16 +0200 Subject: [PATCH] Update dataset.py --- dataset.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/dataset.py b/dataset.py index 12fcc39..aa7b266 100755 --- a/dataset.py +++ b/dataset.py @@ -10,6 +10,12 @@ def audio_converter(audio): if audio.dtype == 'int16': return audio.astype(np.float32, order='C') / 32768.0 + elif audio.dtype == 'int32': + return audio.astype(np.float32, order='C') / (2**31-1) + elif audio.dtype == 'float32': + return audio.astype(np.float32, order='C') + elif audio.dtype == 'float64': + return audio.astype(np.float32, order='C') else: print('unimplemented audio data type conversion...')