blackteay/GPT-SoVITS
1
0
Fork 0
mirror of https://github.com/RVC-Boss/GPT-SoVITS.git synced 2025-04-05 19:41:56 +08:00
Code Issues Packages Projects Releases Wiki Activity

Merge 134098cc4babc9af2f428fb138c8b010e9601130 into 5dfce9a3f0def7f1ee1e075df569b0b2d41df9e3

Browse Source
This commit is contained in:
Yuan-Man 2024-08-21 13:42:26 -07:00 committed by GitHub
commit 4d5bb3042a
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 25 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
GPT_SoVITS/process_dataset.py Normal file
View File

@ -0,0 +1,25 @@
import os
# Supporting third-party datasets with the format where each audio file corresponds to a text file.
# For example, voice01.wav -> voice01.txt.
def convert_dataset(input_folder, output_file, language='zh'):
audio_files = []
for f in os.listdir(input_folder):
if f.endswith('.wav'):
audio_files.append(f)
with open(output_file, 'w', encoding='utf-8') as output:
for audio_file in audio_files:
audio_path = os.path.join(input_folder, audio_file)
text_file = os.path.join(input_folder, audio_file.replace('.wav', '.txt'))
with open(text_file, 'r', encoding='utf-8') as text_content:
text = text_content.read().replace('\n', '')
speaker_name = os.path.splitext(audio_file)[0]
output_line = f'{audio_path}|{speaker_name}|{language}|{text}\n'
output.write(output_line)