| """ from https://github.com/jaywalnut310/glow-tts """ |
|
|
| import numpy as np |
| import torch |
| from .model.monotonic_align.core import maximum_path_c |
|
|
|
|
| def maximum_path(value, mask): |
| """ Cython optimised version. |
| value: [b, t_x, t_y] |
| mask: [b, t_x, t_y] |
| """ |
| value = value * mask |
| device = value.device |
| dtype = value.dtype |
| value = value.data.cpu().numpy().astype(np.float32) |
| path = np.zeros_like(value).astype(np.int32) |
| mask = mask.data.cpu().numpy() |
|
|
| t_x_max = mask.sum(1)[:, 0].astype(np.int32) |
| t_y_max = mask.sum(2)[:, 0].astype(np.int32) |
| maximum_path_c(path, value, t_x_max, t_y_max) |
| return torch.from_numpy(path).to(device=device, dtype=dtype) |
|
|
