(input, kernel_size, sigma)
| 318 | |
| 319 | |
| 320 | def _gaussian_blur2d(input, kernel_size, sigma): |
| 321 | if isinstance(sigma, tuple): |
| 322 | sigma = torch.tensor([sigma], dtype=input.dtype) |
| 323 | else: |
| 324 | sigma = sigma.to(dtype=input.dtype) |
| 325 | |
| 326 | ky, kx = int(kernel_size[0]), int(kernel_size[1]) |
| 327 | bs = sigma.shape[0] |
| 328 | kernel_x = _gaussian(kx, sigma[:, 1].view(bs, 1)) |
| 329 | kernel_y = _gaussian(ky, sigma[:, 0].view(bs, 1)) |
| 330 | out_x = _filter2d(input, kernel_x[..., None, :]) |
| 331 | out = _filter2d(out_x, kernel_y[..., None]) |
| 332 | |
| 333 | return out |
| 334 | |
| 335 | |
| 336 | def export_to_video(video_frames, output_video_path, fps): |
no test coverage detected