/DRPL

Primary LanguagePython

This is the implementation of DRPL:Deep Regression Pair Learning for Multi-Focus Image Fusion.

by Jinxing Li*; Xiaobao Guo*; Guangming Lu; Bob Zhang; Yong Xu; Feng Wu; David Zhang.

DRPL

python 3.6.2 numpy 1.15.0
pytorch 0.4.1 or higher torchvision 0.2.1

test: python test_net.py --load_ckpt ./model/40_net_params.pkl or run: sh test.sh

selected training images: data/selected

Please find the training code in https://github.com/Redaimao/DRPL