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

Rename utils due to name conflict

This commit is contained in:
Eren 2018-07-05 17:02:14 +02:00
parent 0ca9d3d3e0
commit b3ae174607
1 changed files with 2 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
notebooks/utils.py → notebooks/synthesis.py
View File

@ -14,7 +14,7 @@ def create_speech(m, s, CONFIG, use_cuda, ap):
chars_var = torch.from_numpy(seq).unsqueeze(0) chars_var = torch.from_numpy(seq).unsqueeze(0)
if use_cuda: if use_cuda:
chars_var = chars_var.cuda() chars_var = chars_var.cuda()
mel_out, linear_out, alignments, stop_tokens = m.forward(chars_var) mel_out, linear_out, alignments, stop_tokens = m.forward(chars_var.long())
linear_out = linear_out[0].data.cpu().numpy() linear_out = linear_out[0].data.cpu().numpy()
alignment = alignments[0].cpu().data.numpy() alignment = alignments[0].cpu().data.numpy()
spec = ap._denormalize(linear_out) spec = ap._denormalize(linear_out)
@ -34,7 +34,7 @@ def visualize(alignment, spectrogram, stop_tokens, CONFIG):
plt.xlabel("Decoder timestamp", fontsize=label_fontsize) plt.xlabel("Decoder timestamp", fontsize=label_fontsize)
plt.ylabel("Encoder timestamp", fontsize=label_fontsize) plt.ylabel("Encoder timestamp", fontsize=label_fontsize)
plt.colorbar() plt.colorbar()
stop_tokens = stop_tokens.squeeze().detach().to('cpu').numpy() stop_tokens = stop_tokens.squeeze().detach().to('cpu').numpy()
plt.subplot(3, 1, 2) plt.subplot(3, 1, 2)
plt.plot(range(len(stop_tokens)), list(stop_tokens)) plt.plot(range(len(stop_tokens)), list(stop_tokens))
@ -46,5 +46,3 @@ def visualize(alignment, spectrogram, stop_tokens, CONFIG):
plt.ylabel("Hz", fontsize=label_fontsize) plt.ylabel("Hz", fontsize=label_fontsize)
plt.tight_layout() plt.tight_layout()
plt.colorbar() plt.colorbar()