Go to file
2024-06-03 19:48:08 +08:00
utils Add files via upload 2024-06-03 19:36:29 +08:00
dataprocessing.py Add files via upload 2024-06-03 19:36:29 +08:00
net.py Add files via upload 2024-06-03 19:36:29 +08:00
README.md Update README.md 2024-06-03 19:48:08 +08:00
test_IVF.py Add files via upload 2024-06-03 19:36:29 +08:00
train.py Add files via upload 2024-06-03 19:36:29 +08:00

PFCFuse: A Poolformer and CNN fusion network for Infrared-Visible Image Fusion

Poolformer-cnn图像融合框架 The implementation of our paper "PFCFuse: A Poolformer and CNN fusion network for Infrared-Visible Image Fusion".

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