From 55230bb317a8e5fd67b547b9a3a3dda3c6a71692 Mon Sep 17 00:00:00 2001 From: erogol Date: Tue, 16 Jun 2020 18:53:02 +0200 Subject: [PATCH] plot spectrogram transpose fix --- utils/visual.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/utils/visual.py b/utils/visual.py index 3d95c2e3..b4ebec9a 100644 --- a/utils/visual.py +++ b/utils/visual.py @@ -29,11 +29,11 @@ def plot_alignment(alignment, info=None, fig_size=(16, 10), title=None): def plot_spectrogram(spectrogram, ap=None, fig_size=(16, 10)): if isinstance(spectrogram, torch.Tensor): - spectrogram_ = spectrogram.detach().cpu().numpy().squeeze() + spectrogram_ = spectrogram.detach().cpu().numpy().squeeze().T else: - spectrogram_ = spectrogram + spectrogram_ = spectrogram.T if ap is not None: - spectrogram_ = ap._denormalize(spectrogram_.T) # pylint: disable=protected-access + spectrogram_ = ap._denormalize(spectrogram_) # pylint: disable=protected-access fig = plt.figure(figsize=fig_size) plt.imshow(spectrogram_, aspect="auto", origin="lower") plt.colorbar()