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 stft_parameters

This commit is contained in:
Enno Hermann 2023-11-14 10:58:33 +01:00
parent da229f3912
commit f37cc4c028
1 changed files with 4 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -15,6 +15,7 @@ from TTS.utils.audio.numpy_transforms import (
db_to_amp, db_to_amp,
griffin_lim, griffin_lim,
mel_to_spec, mel_to_spec,
millisec_to_length,
spec_to_mel, spec_to_mel,
stft, stft,
) )
@ -209,7 +210,9 @@ class AudioProcessor(object):
# setup stft parameters # setup stft parameters
if hop_length is None: if hop_length is None:
# compute stft parameters from given time values # compute stft parameters from given time values
self.hop_length, self.win_length = self._stft_parameters() self.win_length, self.hop_length = millisec_to_length(
frame_length_ms=self.frame_length_ms, frame_shift_ms=self.frame_shift_ms, sample_rate=self.sample_rate
)
else: else:
# use stft parameters from config file # use stft parameters from config file
self.hop_length = hop_length self.hop_length = hop_length
@ -246,21 +249,6 @@ class AudioProcessor(object):
return AudioProcessor(verbose=verbose, **config.audio) return AudioProcessor(verbose=verbose, **config.audio)
return AudioProcessor(verbose=verbose, **config) return AudioProcessor(verbose=verbose, **config)
### setting up the parameters ###
def _stft_parameters(
self,
) -> Tuple[int, int]:
"""Compute the real STFT parameters from the time values.
Returns:
Tuple[int, int]: hop length and window length for STFT.
"""
factor = self.frame_length_ms / self.frame_shift_ms
assert (factor).is_integer(), " [!] frame_shift_ms should divide frame_length_ms"
hop_length = int(self.frame_shift_ms / 1000.0 * self.sample_rate)
win_length = int(hop_length * factor)
return hop_length, win_length
### normalization ### ### normalization ###
def normalize(self, S: np.ndarray) -> np.ndarray: def normalize(self, S: np.ndarray) -> np.ndarray:
"""Normalize values into `[0, self.max_norm]` or `[-self.max_norm, self.max_norm]` """Normalize values into `[0, self.max_norm]` or `[-self.max_norm, self.max_norm]`