DRL-FPD
%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
This package contains the source code which is associated with the following paper:
Yu Liu, Lei Wang, Huafeng Li, Xun Chen, "Multi-focus image fusion with deep residual learning and focus property detection", Information Fusion, vol. 86-87, p. 1-16, 2022.
Edited by Lei Wang and Yu Liu.
Usage of this code is free for research purposes only.
Thank you.
%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
Requirements:
- CUDA 10
- Python 3.6
- pytorch 1.10.0
- tqdm 4.63.1
- pandas 1.1.5
- torchvision 0.11.0
- imageio 2.15.0
- matplotlib 3.3.4
- scikit-image 0.17.2
- scipy 1.2.1
- opencv-python 4.5.5.64
- numpy 1.19.2
- pillow 8.4.0
The "test" folder contains the test code for the final fusion results.
- "test_Lytro.py" are used to generate the fusion results for the "Lytro" dataset.
- "test_mffw.py" are used to generate the fusion results for the "mffw" dataset.
- "test_natural.py" are used to generate the fusion results for the "natural" dataset.
- "test_multiple.py" are used to generate the fusion results for the "multiple" dataset.
Train the model with "Train.py"
%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
Contact:
Don't hesitate to contact me if you meet any problems when using this code.
Yu Liu
Department of Biomedical Engineering
Hefei University of Technology
Email: yuliu@hfut.edu.cn; lyuxxz@163.com
Homepage: https://sites.google.com/site/yuliu316316; https://github.com/yuliu316316
Last update: 14-Feb-2023