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