Skip to content

yuliu316316/VDMUFusion

Repository files navigation

VDMUFusion

%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%

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

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

 
 
 

Contributors

Languages