pfcfuse/README.md

29 lines
587 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-05-01 15:44:50 +08:00
Poolformer-cnn图像融合框架
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
```