WSJO数据集原始 文件.wv1&.wv2转换成wav文件
最近做语音分离实验需要用到wsj0-2mix数据集,但是从李宏毅语音分离教程里面获取的wsj0-2mix只有一部分。从网上获取到了完整的WSJO数据集后,由于原始的语音文件后缀是wv1或者wv2,创建wsj0-2mix需要wav文件,所以需要通过工具进行转换。网上的教程使用后只能生成一堆空文件夹,无法生成转换后的wav文件。因此,在成功解决这个问题后,把采用的方法分享一下,按照下列步骤能完美进行转换。
准备
- 平台:Windows
- 工具:
- python
- sph2pipe
- 数据集:WSJ0
一、安装sph2pipe
大家按照之前网上的方法只能生成一堆空文件夹就是因为没有安装sph2pipe。按照下列步骤操作
- 下载sph2pipe(https://www.ldc.upenn.edu/language-resources/tools/sphere-conversion-tools),选择版本2.5,2.1在Windows下已经无法运行
-
配置环境变量,如下将sph2pipe.exe的路径加入你电脑的系统变量Path里面,
-
找到你下载的文件,可以看到里面有一个exe文件
- 验证sph2pipe能否成功运行,首先运行cmd或者powershell,保证当前目录里面含有sph2pipe.exe文件,输入sph2pipe,有如下提示就可以了
- 下面就可以使用sph2pipe进行类型转换
三、转换代码
下面就是转换代码,详细说一下你需要改哪些地方,特别是第三点,非常重要!!!
- root_dir:改成你下载的wsj0的根目录文件,代码里面有例子
- my_path:改成你想要保存转换后wav文件存储的路径
- cmd = "E: \sph2pipe_v2.5.tar \sph2pipe_v2.5\sph2pipe -f wav " + speech_dir + " " + target_dir:把E: \sph2pipe_v2.5.tar \sph2pipe_v2.5改成你sph2pipe.exe存在的路径,注意代码中是“\ \”(双斜杠),
"""
# example:
# 11-1.1/wsj0/si_tr_s/01t/01to030v.wv1 is converted to wav and
# stored in YOUR_PATH/wsj0/si_tr_s/01t/01to030v.wav
"""
import os
# 你下载的wsj0的根目录 例子:E:\\csr_1_comp_LDC93S6A\\csr_1_comp,
root_dir = ""
# the disc number
disc_dir = []
for list_disc in os.listdir(root_dir):
if list_disc not in ["text", "11-13.1"]: #doc file and 11-13.1 file do not contain .wv files
# the data dir for each disc
disc_dir.append(os.path.join(root_dir, list_disc, "wsj0"))
# 转换后的文件想要保存的位置
my_path = ""
if not os.path.exists(my_path):
os.mkdir(my_path)
# # the sub_data dir for each disc
for i, list_sub_data in enumerate(disc_dir):
for sub_data_dir in os.listdir(list_sub_data):
if (not sub_data_dir.startswith("si")) and (not sub_data_dir.startswith("sd")):
continue
s_dir = os.path.join(my_path, sub_data_dir)
if not os.path.exists(s_dir):
os.mkdir(s_dir)
if sub_data_dir[0][0] == 's':
datatype_dir = os.path.join(list_sub_data, sub_data_dir)
for list_spk in os.listdir(datatype_dir):
spk_dir = os.path.join(s_dir, list_spk)
spk_dir_abs = os.path.join(datatype_dir, list_spk)
if not os.path.exists(spk_dir):
os.mkdir(spk_dir)
for wv_file in os.listdir(spk_dir_abs):
if (not wv_file.endswith('.wv1')) and (not wv_file.endswith('.wv2')):
continue
speech_dir = os.path.join(spk_dir_abs, wv_file)
if wv_file.split('.')[1] == "wv1":
target_name = wv_file.split(sep='.')[0] + '.wav'
elif wv_file.split('.')[1] == 'wv2':
target_name = wv_file.split(sep='.')[0] + '_1.wav'
target_dir = spk_dir + '\\' + target_name
# 一定要注意!!! sph2pipe -f wav前面的路径必须包含上面讲到的sph2pipe.exe,你只需要根据你sph2pipe.exe存放的位置修改这段路径:E:\\sph2pipe_v2.5.tar\\sph2pipe_v2.5
cmd = "E:\\sph2pipe_v2.5.tar\\sph2pipe_v2.5\\sph2pipe -f wav " + speech_dir + " " + target_dir
os.system(cmd)
四、结果展示
-
编译器窗口:如果你没有按照sph2pipe,终端输出的就是一堆红色乱码,然后生成的文件夹都是空的
-
生成文件,比如我打开第一个sd_dt_05,完美转换
cn-hangzhou.aliyuncs.com/imgs/image-20231129162528077.png" alt=“image-20231129162528077” style=“zoom:50%;” />