[dataset]support aishell3(tested)

4 years ago · 0cc3f569fa
parent 3084fdeb10
commit 0cc3f569fa
2 changed files with 6 additions and 0 deletions
--- a/synthesizer/preprocess.py
+++ b/synthesizer/preprocess.py
@ -19,6 +19,11 @@ data_info = {
        "trans_filepath": "train/TRANS.txt",
        "speak_func": preprocess_speaker_general
    },
+    "aishell3":{
+        "subfolders": ["train/wav"],
+        "trans_filepath": "train/content.txt",
+        "speak_func": preprocess_speaker_general
+    },
 }

 def preprocess_dataset(datasets_root: Path, out_dir: Path, n_processes: int,
--- a/synthesizer_preprocess_audio.py
+++ b/synthesizer_preprocess_audio.py
@ -8,6 +8,7 @@ import argparse
 recognized_datasets = [
    "aidatatang_200zh",
    "magicdata",
+    "aishell3"
 ]

 if __name__ == "__main__":