- create conda environment
conda create -n LDFusion python=3.9.12
conda activate LDFusion
- Install Dependencies
pip install -r requirements.txt
(recommended cuda11.1 and torch 1.8.2)
Please put test data into the test_imgs
directory (infrared images in ir
subfolder, visible images in vi
subfolder), and run python src/test.py
.
(Note: The weight files (*.pt) might require independent download from the repository)
Then, the fused results will be saved in the ./results/
folder.