GST Tacotron

GST-TACOTRON-PYTORCH

Eine Pytorch-Implementierung von Style-Token: unbeaufsichtigte Modellierung, Kontrolle und Übertragung in der End-to-End-Sprachsynthese

Aktualisieren

Fügen Sie Unterstützung für Blizzard -Datensatz hinzu.

Anforderungen

pip3 install -r requirements.txt

Dateistruktur

• Hyperparameters.py --- Hyperparameter
• Network.py --- Encoder und Decoder
• Modules.py --- Einige Module für Tacotron
• Loss.py --- Verlustfunktion
• Data.py --- Datensatzlader
• utils.py --- Einige Util-Funktionen für die Daten i/o
• Synthesis.py --- Sprachgenerierung

Wie man trainiert

• Laden Sie einen Multispeaker -Datensatz herunter
• Preprocess Ihre Daten vorab und implementieren Sie Ihre Funktion get_XX_data in Data.py
• Hyperparameter in Hyperparameters.py einstellen.py
• Machen Sie wie folgt ein Verzeichnis namens log :

 --- log
|    |
|    --- log[log_number]
|
--- code
     |
     --- Tacotron
             |
             --- train.py
             |
             --- Network.py
             |
           ......

• Run Train.py

python3 train.py [log_number] [dataset_size] [start_epoch]

[log_number]: the log directory number
[dataset_size]: int or all
[start_epoch]: which epoch start to train (0 if start from scratch )

for example:
python3 train.py 0 all 0

Wie man WAV erzeugt

Run generate.py . Ersetzen Sie den text in generate.py durch alle chinesischen Sätze, wie Sie möchten

Das vorgelagerte Modell wird auf dem chinesischen Datensatz geschult und unterstützt daher nur Chinesisch.

Sterngeschichte