KernelFilter-PyTorch

A customized operation of PyTorch for implementing Kernel Prediction denoising networks.

See：

• Kernel-Predicting Convolutional Networks for Denoising Monte Carlo Renderings
• Neural Temporal Adaptive Sampling and Denoising

Kernel Filter

KernelFilterClass.forward(input, kernels, dilation=1) → Tensor

Applies a 2D Kernel Filter over an input image.

For an input size (N, C, H, W), the size of kernels must be (N, K_Size x K_Size, H, W). The output size is the same as input size.

Letting denote the value at position p, channel c of the input image and denote the value at position p, channel n of the kernels, the Kernel Filter can be defined as:

Compile

bash ./install_kernel_filter.sh