File size: 816 Bytes
f8ff705 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 |
import soundfile as sf
import os
os.makedirs("dataset", exist_ok=True)
archive_path = "test"
wav_dir = os.path.join(archive_path, "wav")
segments_file = os.path.join(archive_path, "text.all")
with open(segments_file, "r", encoding="utf-8") as f:
for _id, line in enumerate(f):
segment = line.split(" ")[0]
text = " ".join(line.split(" ")[1:])
wav_name, _, time = segment.split("_")
time = time.replace("seg-", "")
start, stop = time.split(":")
start = int(int(start) / 100 * 16_000)
stop = int(int(stop) / 100 * 16_000)
wav_path = os.path.join(wav_dir, wav_name + ".wav")
sound, _ = sf.read(wav_path, start=start, stop=stop)
sf.write(f"dataset/{segment}.wav", sound, 16_000)
open(f"dataset/{segment}.txt", "w").write(text)
|