From 0856ea2ebdcb976db6e786de5cd79fae3d35cd4c Mon Sep 17 00:00:00 2001
From: zhifu gao <zhifu.gzf@alibaba-inc.com>
Date: 星期一, 20 二月 2023 18:18:35 +0800
Subject: [PATCH] Merge pull request #136 from alibaba-damo-academy/dev_cmz
---
funasr/utils/wav_utils.py | 12 +++++++++---
1 files changed, 9 insertions(+), 3 deletions(-)
diff --git a/funasr/utils/wav_utils.py b/funasr/utils/wav_utils.py
index afc0ec9..590ccdb 100644
--- a/funasr/utils/wav_utils.py
+++ b/funasr/utils/wav_utils.py
@@ -298,14 +298,20 @@
os.rename(text_file, "{}.bak".format(text_file))
wav_dict = {}
for line in wav_lines:
- sample_name, wav_path = line.strip().split()
+ parts = line.strip().split()
+ if len(parts) < 2:
+ continue
+ sample_name, wav_path = parts
wav_dict[sample_name] = wav_path
text_dict = {}
for line in text_lines:
- sample_name, txt = line.strip().split(" ", 1)
+ parts = line.strip().split(" ", 1)
+ if len(parts) < 2:
+ continue
+ sample_name, txt = parts
text_dict[sample_name] = txt
filter_count = 0
- with open(wav_file) as f_wav, open(text_file) as f_text:
+ with open(wav_file, "w") as f_wav, open(text_file, "w") as f_text:
for sample_name, wav_path in wav_dict.items():
if sample_name in text_dict.keys():
f_wav.write(sample_name + " " + wav_path + "\n")
--
Gitblit v1.9.1