| 28 | |
| 29 | |
| 30 | def dur_to_figure(dur_gt, dur_pred, txt): |
| 31 | dur_gt = dur_gt.long().cpu().numpy() |
| 32 | dur_pred = dur_pred.long().cpu().numpy() |
| 33 | dur_gt = np.cumsum(dur_gt) |
| 34 | dur_pred = np.cumsum(dur_pred) |
| 35 | fig = plt.figure(figsize=(12, 6)) |
| 36 | for i in range(len(dur_gt)): |
| 37 | shift = (i % 8) + 1 |
| 38 | plt.text(dur_gt[i], shift, txt[i]) |
| 39 | plt.text(dur_pred[i], 10 + shift, txt[i]) |
| 40 | plt.vlines(dur_gt[i], 0, 10, colors='b') # blue is gt |
| 41 | plt.vlines(dur_pred[i], 10, 20, colors='r') # red is pred |
| 42 | return fig |
| 43 | |
| 44 | |
| 45 | def f0_to_figure(f0_gt, f0_cwt=None, f0_pred=None): |