Medix-AI/coqui-tts
Medix-AI
/
coqui-tts
mirror of https://github.com/coqui-ai/TTS.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

refactor(audio.processor): remove duplicate build_mel_basis

This commit is contained in:
Enno Hermann 2023-11-14 10:49:26 +01:00
parent 754877784b
commit da229f3912
1 changed files with 8 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
TTS/utils/audio/processor.py
View File

@ -10,6 +10,7 @@ import soundfile as sf
from TTS.tts.utils.helpers import StandardScaler from TTS.tts.utils.helpers import StandardScaler
from TTS.utils.audio.numpy_transforms import ( from TTS.utils.audio.numpy_transforms import (
amp_to_db, amp_to_db,
build_mel_basis,
compute_f0, compute_f0,
db_to_amp, db_to_amp,
griffin_lim, griffin_lim,
@ -223,7 +224,13 @@ class AudioProcessor(object):
for key, value in members.items(): for key, value in members.items():
print(" | > {}:{}".format(key, value)) print(" | > {}:{}".format(key, value))
# create spectrogram utils # create spectrogram utils
self.mel_basis = self._build_mel_basis() self.mel_basis = build_mel_basis(
sample_rate=self.sample_rate,
fft_size=self.fft_size,
num_mels=self.num_mels,
mel_fmax=self.mel_fmax,
mel_fmin=self.mel_fmin,
)
# setup scaler # setup scaler
if stats_path and signal_norm: if stats_path and signal_norm:
mel_mean, mel_std, linear_mean, linear_std, _ = self.load_stats(stats_path) mel_mean, mel_std, linear_mean, linear_std, _ = self.load_stats(stats_path)
@ -240,20 +247,6 @@ class AudioProcessor(object):
return AudioProcessor(verbose=verbose, **config) return AudioProcessor(verbose=verbose, **config)
### setting up the parameters ### ### setting up the parameters ###
def _build_mel_basis(
self,
) -> np.ndarray:
"""Build melspectrogram basis.
Returns:
np.ndarray: melspectrogram basis.
"""
if self.mel_fmax is not None:
assert self.mel_fmax <= self.sample_rate // 2
return librosa.filters.mel(
sr=self.sample_rate, n_fft=self.fft_size, n_mels=self.num_mels, fmin=self.mel_fmin, fmax=self.mel_fmax
)
def _stft_parameters( def _stft_parameters(
self, self,
) -> Tuple[int, int]: ) -> Tuple[int, int]: