/ZMFF

Primary LanguagePython

ZMFF

This is an implementation of the following paper.

@article{Hu2023ZMFF,
title = {ZMFF: Zero-shot multi-focus image fusion},
author={Hu, Xingyu and Jiang, Junjun and Liu, Xianming and Ma, Jiayi},
journal = {Information Fusion},
volume = {92},
pages = {127-138},
year = {2023},
}

Preparation

Dependencies

  • python >= 3.6
  • pytorch >= 1.5.0
  • NVIDIA GPU
  • CUDA

Data Preparation

Testing

For testing, please run:

python main.py

results

The output results will be stored in ./result.

Our results in Lytro, Real-MFF, MFI-WHU datasets can be downloaded.