| 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) |
|
|
