(kernel_size: int, sigma: float)
| 135 | |
| 136 | |
| 137 | def gaussian_kernel(kernel_size: int, sigma: float): |
| 138 | x = torch.linspace(-(kernel_size - 1) / 2, (kernel_size - 1) / 2, |
| 139 | kernel_size) |
| 140 | gauss = torch.exp(-x**2 / (2 * sigma**2)) |
| 141 | return gauss / gauss.sum() |
| 142 | |
| 143 | |
| 144 | def smooth_motion_rep(motion_rep, kernel_size: int, sigma: float): |