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

allow choosing the log function used for amptodb conversion

This commit is contained in:
Eren Gölge 2021-04-05 00:01:38 +02:00
parent 718f6e1568
commit e0b3008c31
1 changed files with 13 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
TTS/utils/audio.py
View File

@ -47,6 +47,7 @@ class AudioProcessor(object):
sample_rate=None, sample_rate=None,
resample=False, resample=False,
num_mels=None, num_mels=None,
log_func='np.log10',
min_level_db=None, min_level_db=None,
frame_shift_ms=None, frame_shift_ms=None,
frame_length_ms=None, frame_length_ms=None,
@ -96,6 +97,15 @@ class AudioProcessor(object):
self.trim_db = trim_db self.trim_db = trim_db
self.do_sound_norm = do_sound_norm self.do_sound_norm = do_sound_norm
self.stats_path = stats_path self.stats_path = stats_path
# setup exp_func for db to amp conversion
print(f'self.log_func = {log_func}')
exec(f'self.log_func = {log_func}') #pylint: disable=exec-used
if self.log_func.__name__ == 'log':
self.exp_func = np.exp
elif self.log_func.__name__ == 'log10':
self.exp_func = lambda x: 10 ** x
else:
raise ValueError(' [!] unknown `log_func` value.')
# 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
@ -227,11 +237,12 @@ class AudioProcessor(object):
### DB and AMP conversion ### ### DB and AMP conversion ###
# pylint: disable=no-self-use # pylint: disable=no-self-use
def _amp_to_db(self, x): def _amp_to_db(self, x):
return self.spec_gain * np.log10(np.maximum(1e-5, x)) return self.spec_gain * self.log_func(np.maximum(1e-5, x))
# pylint: disable=no-self-use # pylint: disable=no-self-use
def _db_to_amp(self, x): def _db_to_amp(self, x):
return np.power(10.0, x / self.spec_gain) return self.exp_func(x / self.spec_gain)
### Preemphasis ### ### Preemphasis ###
def apply_preemphasis(self, x): def apply_preemphasis(self, x):