mirror of https://github.com/coqui-ai/TTS.git
Fix imports
This commit is contained in:
parent
16b70be0dd
commit
9f23ad6a0f
|
@ -6,7 +6,6 @@ import torch
|
||||||
from coqpit import Coqpit
|
from coqpit import Coqpit
|
||||||
from torch import nn
|
from torch import nn
|
||||||
|
|
||||||
from TTS.utils.audio import AudioProcessor
|
|
||||||
|
|
||||||
# pylint: skip-file
|
# pylint: skip-file
|
||||||
|
|
||||||
|
|
|
@ -308,14 +308,14 @@ def mls(root_path, meta_files=None):
|
||||||
# ======================================== VOX CELEB ===========================================
|
# ======================================== VOX CELEB ===========================================
|
||||||
def voxceleb2(root_path, meta_file=None):
|
def voxceleb2(root_path, meta_file=None):
|
||||||
"""
|
"""
|
||||||
:param meta_file Used only for consistency with load_meta_data api
|
:param meta_file Used only for consistency with load_tts_samples api
|
||||||
"""
|
"""
|
||||||
return _voxcel_x(root_path, meta_file, voxcel_idx="2")
|
return _voxcel_x(root_path, meta_file, voxcel_idx="2")
|
||||||
|
|
||||||
|
|
||||||
def voxceleb1(root_path, meta_file=None):
|
def voxceleb1(root_path, meta_file=None):
|
||||||
"""
|
"""
|
||||||
:param meta_file Used only for consistency with load_meta_data api
|
:param meta_file Used only for consistency with load_tts_samples api
|
||||||
"""
|
"""
|
||||||
return _voxcel_x(root_path, meta_file, voxcel_idx="1")
|
return _voxcel_x(root_path, meta_file, voxcel_idx="1")
|
||||||
|
|
||||||
|
|
|
@ -110,10 +110,10 @@ class SpeakerManager:
|
||||||
|
|
||||||
@staticmethod
|
@staticmethod
|
||||||
def parse_speakers_from_data(items: list) -> Tuple[Dict, int]:
|
def parse_speakers_from_data(items: list) -> Tuple[Dict, int]:
|
||||||
"""Parse speaker IDs from data samples retured by `load_meta_data()`.
|
"""Parse speaker IDs from data samples retured by `load_tts_samples()`.
|
||||||
|
|
||||||
Args:
|
Args:
|
||||||
items (list): Data sampled returned by `load_meta_data()`.
|
items (list): Data sampled returned by `load_tts_samples()`.
|
||||||
|
|
||||||
Returns:
|
Returns:
|
||||||
Tuple[Dict, int]: speaker IDs and number of speakers.
|
Tuple[Dict, int]: speaker IDs and number of speakers.
|
||||||
|
@ -127,7 +127,7 @@ class SpeakerManager:
|
||||||
"""Set speaker IDs from data samples.
|
"""Set speaker IDs from data samples.
|
||||||
|
|
||||||
Args:
|
Args:
|
||||||
items (List): Data sampled returned by `load_meta_data()`.
|
items (List): Data sampled returned by `load_tts_samples()`.
|
||||||
"""
|
"""
|
||||||
self.speaker_ids, _ = self.parse_speakers_from_data(items)
|
self.speaker_ids, _ = self.parse_speakers_from_data(items)
|
||||||
|
|
||||||
|
|
|
@ -5,6 +5,7 @@ from torch.utils.data import Dataset
|
||||||
|
|
||||||
from TTS.utils.audio import AudioProcessor
|
from TTS.utils.audio import AudioProcessor
|
||||||
from TTS.vocoder.datasets.gan_dataset import GANDataset
|
from TTS.vocoder.datasets.gan_dataset import GANDataset
|
||||||
|
from TTS.vocoder.datasets.preprocess import load_wav_data, load_wav_feat_data
|
||||||
from TTS.vocoder.datasets.wavegrad_dataset import WaveGradDataset
|
from TTS.vocoder.datasets.wavegrad_dataset import WaveGradDataset
|
||||||
from TTS.vocoder.datasets.wavernn_dataset import WaveRNNDataset
|
from TTS.vocoder.datasets.wavernn_dataset import WaveRNNDataset
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue