Skip to content

Latest commit

 

History

History
26 lines (16 loc) · 1.32 KB

File metadata and controls

26 lines (16 loc) · 1.32 KB

KernelFilter-PyTorch

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

3x3_kernel_filter_demonstration

See:

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