2024-05-01 15:46:07 +08:00
|
|
|
# PFCFuse: A Poolformer and CNN fusion network for Infrared-Visible Image Fusion
|
2024-06-03 19:42:00 +08:00
|
|
|
The implementation of our paper "PFCFuse: A Poolformer and CNN fusion network for Infrared-Visible Image Fusion".
|
|
|
|
## Recommended Environment:
|
2024-06-03 19:48:08 +08:00
|
|
|
python=3.8\
|
|
|
|
torch=1.12.1+cu113\
|
|
|
|
scipy=1.9.3\
|
|
|
|
scikit-image=0.19.2\
|
|
|
|
scikit-learn=1.1.3\
|
|
|
|
tqdm=4.62.0
|
|
|
|
## Network Architecture:
|
|
|
|
Our PFCFuse is implemented in ``net.py``.
|
|
|
|
## Training:
|
|
|
|
### Data preprocessing
|
|
|
|
Run
|
|
|
|
```
|
|
|
|
python dataprocessing.py
|
|
|
|
```
|
|
|
|
### Model training
|
|
|
|
Run
|
|
|
|
```
|
|
|
|
python train.py
|
|
|
|
```
|
|
|
|
## Testing:
|
|
|
|
Run
|
|
|
|
```
|
|
|
|
python test_IVF.py
|
|
|
|
```
|
2024-06-26 15:32:37 +08:00
|
|
|
|
|
|
|
## 相关工作
|
|
|
|
@inproceedings{zhao2023cddfuse,
|
|
|
|
title={Cddfuse: Correlation-driven dual-branch feature decomposition for multi-modality image fusion},
|
|
|
|
author={Zhao, Zixiang and Bai, Haowen and Zhang, Jiangshe and Zhang, Yulun and Xu, Shuang and Lin, Zudi and Timofte, Radu and Van Gool, Luc},
|
|
|
|
booktitle={Proceedings of the IEEE/CVF conference on computer vision and pattern recognition},
|
|
|
|
pages={5906--5916},
|
|
|
|
year={2023}
|
|
|
|
}
|