From 569fa31d06b694d4bb11569aaebe29b646f18287 Mon Sep 17 00:00:00 2001 From: shahules786 Date: Tue, 23 Aug 2022 13:34:26 +0530 Subject: [PATCH] reshape read audio --- enhancer/utils/io.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/enhancer/utils/io.py b/enhancer/utils/io.py index 510ab86..04015e6 100644 --- a/enhancer/utils/io.py +++ b/enhancer/utils/io.py @@ -29,6 +29,8 @@ class Audio: if os.path.exists(audio): audio,sampling_rate = librosa.load(audio,sr=sampling_rate,mono=False, offset=offset,duration=duration) + if len(audio.shape) == 1: + audio = audio.reshape(1,-1) else: raise FileNotFoundError(f"File {audio} deos not exist") elif isinstance(audio,np.ndarray):