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

Audio.py check and remove redundant code

This commit is contained in:
Eren G 2018-08-09 15:12:53 +02:00
parent ecd31af125
commit f1271b4ce8
2 changed files with 19 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
config.json
View File

@ -21,7 +21,7 @@
"eval_batch_size":-1, "eval_batch_size":-1,
"r": 5, "r": 5,
"griffin_lim_iters": 50, "griffin_lim_iters": 60,
"power": 1.5, "power": 1.5,
"num_loader_workers": 8, "num_loader_workers": 8,

29
utils/audio.py
View File

@ -66,8 +66,6 @@ class AudioProcessor(object):
n_fft = (self.num_freq - 1) * 2 n_fft = (self.num_freq - 1) * 2
hop_length = int(self.frame_shift_ms / 1000.0 * self.sample_rate) hop_length = int(self.frame_shift_ms / 1000.0 * self.sample_rate)
win_length = int(self.frame_length_ms / 1000.0 * self.sample_rate) win_length = int(self.frame_length_ms / 1000.0 * self.sample_rate)
hop_length = 256
win_length = 1024
print(" | > fft size: {}, hop length: {}, win length: {}".format( print(" | > fft size: {}, hop length: {}, win length: {}".format(
n_fft, hop_length, win_length)) n_fft, hop_length, win_length))
return n_fft, hop_length, win_length return n_fft, hop_length, win_length
@ -107,17 +105,26 @@ class AudioProcessor(object):
else: else:
return self._griffin_lim(S**self.power) return self._griffin_lim(S**self.power)
# def _griffin_lim(self, S):
# '''Applies Griffin-Lim's raw.
# '''
# S_best = copy.deepcopy(S)
# for i in range(self.griffin_lim_iters):
# S_t = self._istft(S_best)
# est = self._stft(S_t)
# phase = est / np.maximum(1e-8, np.abs(est))
# S_best = S * phase
# S_t = self._istft(S_best)
# y = np.real(S_t)
# return y
def _griffin_lim(self, S): def _griffin_lim(self, S):
'''Applies Griffin-Lim's raw. angles = np.exp(2j * np.pi * np.random.rand(*S.shape))
''' S_complex = np.abs(S).astype(np.complex)
S_best = copy.deepcopy(S) y = self._istft(S_complex * angles)
for i in range(self.griffin_lim_iters): for i in range(self.griffin_lim_iters):
S_t = self._istft(S_best) angles = np.exp(1j * np.angle(self._stft(y)))
est = self._stft(S_t) y = self._istft(S_complex * angles)
phase = est / np.maximum(1e-8, np.abs(est))
S_best = S * phase
S_t = self._istft(S_best)
y = np.real(S_t)
return y return y
def melspectrogram(self, y): def melspectrogram(self, y):