reshape read audio

2022-08-23 13:34:26 +05:30 · 2022-08-23 13:34:26 +05:30 · 569fa31d06
parent 65540148f7
commit 569fa31d06
1 changed files with 2 additions and 0 deletions
--- 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):