From 0cc3f569fa5701a3885a2ba52b7d2201c2f32ae0 Mon Sep 17 00:00:00 2001
From: Nemo <956090487@qq.com>
Date: Wed, 25 Aug 2021 23:11:29 +0800
Subject: [PATCH] [dataset]support aishell3(tested)

---
 synthesizer/preprocess.py       | 5 +++++
 synthesizer_preprocess_audio.py | 1 +
 2 files changed, 6 insertions(+)

diff --git a/synthesizer/preprocess.py b/synthesizer/preprocess.py
index 60fd2cc..344a74d 100644
--- 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,
diff --git a/synthesizer_preprocess_audio.py b/synthesizer_preprocess_audio.py
index 2af7b9f..871f5e7 100644
--- 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__":