Merge pull request #4 from Jackwaterveg/cli_infer

revise
3 years ago · 2135aeb08d
parent 8828210f72 8ec576f477
commit 2135aeb08d
1 changed files with 6 additions and 1 deletions
--- a/paddlespeech/cli/asr/infer.py
+++ b/paddlespeech/cli/asr/infer.py
@ -228,10 +228,15 @@ class ASRExecutor(BaseExecutor):
                    audio = audio.mean(axis=1)
                else:
                    audio = audio[:, 0]
+                # pcm16 -> pcm 32
                audio = audio.astype("float32")
+                bits = np.iinfo(np.int16).bits
+                audio = audio / (2**(bits - 1))
                audio = librosa.resample(audio, audio_sample_rate,
                                         self.sample_rate)
                audio_sample_rate = self.sample_rate
+                # pcm32 -> pcm 16
+                audio = audio * (2**(bits - 1))
                audio = np.round(audio).astype("int16")
            else:
                audio = audio[:, 0]