%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
This package contains the source code which is associated with the following paper:
Yu Shi, Yu Liu, Juan Cheng, Z. Jane. Wang, and Xun Chen, "VDMUFusion: A Versatile Diffusion Model-Based Unsupervised Framework for Image Fusion", submitted to IEEE Transactions on Image Processing, 2024. (under review, second round)
Edited by Yu Shi.
Usage of this code is free for research purposes only.
The VDMUFusion model is learned by pytorch. This package contains our pre-trained models and code, which you can use directly for testing. However, if you wish to retrain the models, you will need to download the datasets for multiple image fusion tasks and place them in the corresponding file paths.
Thank you.
%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
Requirements:
- CUDA 10
- Python 3.7
- pytorch 1.7.1
- Torchvison 0.8.2
- numpy 1.19.2
%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
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: 30-Sep-2024