GST Tacotron

GST-Tacotron-Pytorch

Реализация токенов стиля Pytorch: моделирование, контроль и передача стиля неконтролируемого стиля при синтезе сквозного речи

Обновлять

Добавьте поддержку набора данных Blizzard.

Требования

pip3 install -r requirements.txt

Структура файла

• Hyperparameters.py --- Гиперпараметры
• Network.py --- энкодер и декодер
• Modules.py --- Некоторые модули для такотрона
• Loss.py --- Функция потерь
• Data.py --- Загрузчик набора данных
• utils.py --- Некоторые функции UTIL для ввода-вывода данных
• Synthesis.py --- Речь генерация

Как тренироваться

• Загрузите набор данных Multipeaker
• Предварительно обработать ваши данные и реализовать функцию get_XX_data в Data.py
• Установить гиперпараметры в Hyperparameters.py
• Сделайте каталог с именем log следующим образом:

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

• Запустить 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

Как генерировать wav

Запустить generate.py . Замените text в generate.py на любые китайские предложения, как вам нравится, прежде чем запустить

Предоставленная модель, предоставленная, обучается китайскому набору данных, поэтому теперь она поддерживает только китайский.

Звездная история