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
|
|
|
|
```
|