cvlab-yonsei / dkn

An official implementation of "Deformable Kernel Network for Joint Image Filtering" (IJCV 2020) in PyTorch.

Deformable Kernel Network for Joint Image Filtering

This is the pytorch implementation of our DKN [Paper]

1. Requirement

Pytorch 1.0.0

Cuda 10.0

Opencv 3.4

tqdm

logging

2. Dataset

NYU v2 dataset, our split can be downloaded: http://gofile.me/3G5St/2lFq5R3TL

3. Inference

• DKN

python inference.py --rgb images/0_rgb.png --depth images/0_lr.png --k 3 --d 15 --scale 8 --parameter parameter/DKN_8x --output images/result_dkn.png --model DKN

• FDKN

python inference.py --rgb images/0_rgb.png --depth images/0_lr.png --k 3 --d 15 --scale 8 --parameter parameter/FDKN_8x --output images/result_fdkn.png --model FDKN

4. Train

• DKN

python train.py --k 3 --d 15 --scale 8 --model DKN

• FDKN

python train.py --k 3 --d 15 --scale 8 --model FDKN