wassname/TTS
1
0
Fork 0
mirror of https://github.com/wassname/TTS.git synced 2026-09-09 11:16:00 +08:00
Code Issues Packages Projects Releases Wiki Activity

tensorboardx plotting figures

Browse Source
This commit is contained in:
Eren
2018-08-11 16:53:09 +02:00
parent b65f5bd1d0
commit f7add3c8e5
2 changed files with 10 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files
utils/visual.py
+2 -10
View File
@@ -15,11 +15,7 @@ def plot_alignment(alignment, info=None):
plt.xlabel(xlabel)
plt.ylabel('Encoder timestep')
plt.tight_layout()
fig.canvas.draw()
data = np.fromstring(fig.canvas.tostring_rgb(), dtype=np.uint8, sep='')
data = data.reshape((3, ) + fig.canvas.get_width_height()[::-1])
plt.close()
return data
return fig
def plot_spectrogram(linear_output, audio):
@@ -28,8 +24,4 @@ def plot_spectrogram(linear_output, audio):
plt.imshow(spectrogram.T, aspect="auto", origin="lower")
plt.colorbar()
plt.tight_layout()
fig.canvas.draw()
data = np.fromstring(fig.canvas.tostring_rgb(), dtype=np.uint8, sep='')
data = data.reshape((3, ) + fig.canvas.get_width_height()[::-1])
plt.close()
return data
return fig