pfcfuse/README.md

39 lines
995 B
Markdown
Raw Normal View History

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
## 相关工作
2024-06-29 10:53:11 +08:00
```
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}
}
2024-06-29 10:53:11 +08:00
```