From c644ac8f58895b9e29e9cfca79465fd2c0efaa5a Mon Sep 17 00:00:00 2001
From: 游雁 <zhifu.gzf@alibaba-inc.com>
Date: 星期二, 21 十一月 2023 14:09:01 +0800
Subject: [PATCH] funasr v2 setup
---
funasr/bin/ss_inference_launch.py | 9 +++++++--
1 files changed, 7 insertions(+), 2 deletions(-)
diff --git a/funasr/bin/ss_inference_launch.py b/funasr/bin/ss_inference_launch.py
index 64503a0..0c02419 100644
--- a/funasr/bin/ss_inference_launch.py
+++ b/funasr/bin/ss_inference_launch.py
@@ -13,7 +13,7 @@
import numpy as np
import torch
-import soundfile as sf
+import librosa
from funasr.build_utils.build_streaming_iterator import build_streaming_iterator
from funasr.torch_utils.set_all_random_seed import set_all_random_seed
from funasr.utils import config_argparse
@@ -104,7 +104,12 @@
ss_results = speech_separator(**batch)
for spk in range(num_spks):
- sf.write(os.path.join(output_path, keys[0] + '_s' + str(spk+1)+'.wav'), ss_results[spk], sample_rate)
+ # sf.write(os.path.join(output_path, keys[0] + '_s' + str(spk+1)+'.wav'), ss_results[spk], sample_rate)
+ try:
+ librosa.output.write_wav(os.path.join(output_path, keys[0] + '_s' + str(spk+1)+'.wav'), ss_results[spk], sample_rate)
+ except:
+ print("To write wav by librosa, you should install librosa<=0.8.0")
+ raise
torch.cuda.empty_cache()
return ss_results
--
Gitblit v1.9.1