SMFuse
March 1, 2021 · View on GitHub
The code of “SMFuse: Multi-focus Image Fusion via Self-supervised Mask-optimization”
To train :
Run "CUDA_VISIBLE_DEVICES=0 python main.py" to train the network.
To test :
Run "CUDA_VISIBLE_DEVICES=0 python test_one_image.py" to test the network.