From 3bbeb43f5770adf1fb310c24bba2fd017b3a79c7 Mon Sep 17 00:00:00 2001 From: erogol Date: Tue, 17 Mar 2020 13:28:15 +0100 Subject: [PATCH] visualization updates wrt mean-var scaling --- utils/visual.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/utils/visual.py b/utils/visual.py index 1cb9ac5d..b0db7b04 100644 --- a/utils/visual.py +++ b/utils/visual.py @@ -32,22 +32,22 @@ def plot_spectrogram(linear_output, audio, fig_size=(16, 10)): linear_output_ = linear_output.detach().cpu().numpy().squeeze() else: linear_output_ = linear_output - spectrogram = audio._denormalize(linear_output_) # pylint: disable=protected-access + spectrogram = audio._denormalize(linear_output_.T) # pylint: disable=protected-access fig = plt.figure(figsize=fig_size) - plt.imshow(spectrogram.T, aspect="auto", origin="lower") + plt.imshow(spectrogram, aspect="auto", origin="lower") plt.colorbar() plt.tight_layout() return fig -def visualize(alignment, spectrogram_postnet, stop_tokens, text, hop_length, CONFIG, spectrogram=None, output_path=None): +def visualize(alignment, spectrogram_postnet, stop_tokens, text, hop_length, CONFIG, spectrogram=None, output_path=None, figsize=[8, 24]): if spectrogram is not None: num_plot = 4 else: num_plot = 3 label_fontsize = 16 - fig = plt.figure(figsize=(8, 24)) + fig = plt.figure(figsize=figsize) plt.subplot(num_plot, 1, 1) plt.imshow(alignment.T, aspect="auto", origin="lower", interpolation=None)