Mirrors/TTS
1
0
Fork 0
mirror of https://github.com/coqui-ai/TTS.git synced 2026-07-13 05:50:18 +02:00
Code Issues Packages Projects Releases Wiki Activity

update imports for convert* scripts

Browse Source
This commit is contained in:
erogol
2020-07-17 13:14:34 +02:00
parent 9ce4126482
commit a0f488136a
5 changed files with 26 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
TTS/bin/compute_statistics.py
View File

@@ -8,8 +8,8 @@ import numpy as np
from tqdm import tqdm
from TTS.tts.datasets.preprocess import load_meta_data
from TTS.tts.utils.io import load_config
from TTS.tts.utils.audio import AudioProcessor
from TTS.utils.io import load_config
from TTS.utils.audio import AudioProcessor
def main():
"""Run preprocessing process."""

2
TTS/bin/convert_melgan_tflite.py
View File

@@ -2,7 +2,7 @@
import argparse
from TTS.tts.utils.io import load_config
from TTS.utils.io import load_config
from TTS.vocoder.tf.utils.generic_utils import setup_generator
from TTS.vocoder.tf.utils.io import load_checkpoint
from TTS.vocoder.tf.utils.tflite import convert_melgan_to_tflite

2
TTS/bin/convert_melgan_torch_to_tf.py
View File

@@ -6,7 +6,7 @@ import tensorflow as tf
import torch
from fuzzywuzzy import fuzz
from TTS.tts.utils.io import load_config
from TTS.utils.io import load_config
from TTS.vocoder.tf.utils.convert_torch_to_tf_utils import (
compare_torch_tf, convert_tf_name, transfer_weights_torch_to_tf)
from TTS.vocoder.tf.utils.generic_utils import \

8
TTS/bin/convert_tacotron2_tflite.py
View File

@@ -2,11 +2,11 @@
import argparse
from TTS.tts.utils.io import load_config
from TTS.utils.io import load_config
from TTS.tts.utils.text.symbols import symbols, phonemes
from TTS.tf.utils.generic_utils import setup_model
from TTS.tf.utils.io import load_checkpoint
from TTS.tf.utils.tflite import convert_tacotron2_to_tflite
from TTS.tts.tf.utils.generic_utils import setup_model
from TTS.tts.tf.utils.io import load_checkpoint
from TTS.tts.tf.utils.tflite import convert_tacotron2_to_tflite
parser = argparse.ArgumentParser()

31
TTS/bin/convert_tacotron2_torch_to_tf.py
View File

@@ -1,21 +1,29 @@
# %%
import sys
sys.path.append('/home/erogol/Projects')
import os
os.environ['CUDA_VISIBLE_DEVICES'] = ''
# %%
import argparse
import os
import sys
# %%
# print variable match
from pprint import pprint
import numpy as np
import torch
import tensorflow as tf
import torch
from fuzzywuzzy import fuzz
from TTS.tts.utils.text.symbols import phonemes, symbols
from TTS.tts.utils.generic_utils import setup_model
from TTS.tts.utils.io import load_config
from TTS.tf.models.tacotron2 import Tacotron2
from TTS.tf.utils.convert_torch_to_tf_utils import compare_torch_tf, tf_create_dummy_inputs, transfer_weights_torch_to_tf, convert_tf_name
from TTS.tf.utils.generic_utils import save_checkpoint
from TTS.tts.utils.text.symbols import phonemes, symbols
from TTS.utils.io import load_config
from TTS.tts.tf.models.tacotron2 import Tacotron2
from TTS.tts.tf.utils.convert_torch_to_tf_utils import (
compare_torch_tf, convert_tf_name, tf_create_dummy_inputs,
transfer_weights_torch_to_tf)
from TTS.tts.tf.utils.generic_utils import save_checkpoint
sys.path.append('/home/erogol/Projects')
os.environ['CUDA_VISIBLE_DEVICES'] = ''
parser = argparse.ArgumentParser()
parser.add_argument('--torch_model_path',
@@ -108,9 +116,6 @@ for tf_name in tf_var_names:
del torch_var_names[max_idx]
var_map.append((tf_name, matching_name))
# %%
# print variable match
from pprint import pprint
pprint(var_map)
pprint(torch_var_names)