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

fixing size mismatch

Browse Source
This commit is contained in:
Eren
2018-08-10 18:48:43 +02:00
parent 9100e5762a
commit 3b2654203d
3 changed files with 45 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files
utils/visual.py
+2 -2
View File
@@ -17,7 +17,7 @@ def plot_alignment(alignment, info=None):
plt.tight_layout()
fig.canvas.draw()
data = np.fromstring(fig.canvas.tostring_rgb(), dtype=np.uint8, sep='')
data = data.reshape(fig.canvas.get_width_height()[::-1] + (3, ))
data = data.reshape((3, ) + fig.canvas.get_width_height()[::-1])
plt.close()
return data
@@ -30,6 +30,6 @@ def plot_spectrogram(linear_output, audio):
plt.tight_layout()
fig.canvas.draw()
data = np.fromstring(fig.canvas.tostring_rgb(), dtype=np.uint8, sep='')
data = data.reshape(fig.canvas.get_width_height()[::-1] + (3, ))
data = data.reshape((3, ) + fig.canvas.get_width_height()[::-1])
plt.close()
return data