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

convert float16 to float32 for plotting spectrograms

pull/10/head
erogol 2020-11-25 14:50:28 +01:00
parent 7541d2ecaa
commit a1e4ee18f9
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
TTS/tts/utils/visual.py
View File

@ -1,6 +1,8 @@
import torch
import librosa import librosa
import matplotlib import matplotlib
import numpy as np
import torch
matplotlib.use('Agg') matplotlib.use('Agg')
import matplotlib.pyplot as plt import matplotlib.pyplot as plt
from TTS.tts.utils.text import phoneme_to_sequence, sequence_to_phoneme from TTS.tts.utils.text import phoneme_to_sequence, sequence_to_phoneme
@ -43,6 +45,8 @@ def plot_spectrogram(spectrogram,
spectrogram_ = spectrogram.detach().cpu().numpy().squeeze().T spectrogram_ = spectrogram.detach().cpu().numpy().squeeze().T
else: else:
spectrogram_ = spectrogram.T spectrogram_ = spectrogram.T
spectrogram_ = spectrogram_.astype(
np.float32) if spectrogram_.dtype == np.float16 else spectrogram_
if ap is not None: if ap is not None:
spectrogram_ = ap._denormalize(spectrogram_) # pylint: disable=protected-access spectrogram_ = ap._denormalize(spectrogram_) # pylint: disable=protected-access
fig = plt.figure(figsize=fig_size) fig = plt.figure(figsize=fig_size)