From 1ad1204210e5877e8e77782f8cb4d7c9fecd12b6 Mon Sep 17 00:00:00 2001 From: HXY13 <63409863+HXY13@users.noreply.github.com> Date: Mon, 3 Jun 2024 19:48:08 +0800 Subject: [PATCH] Update README.md --- README.md | 26 ++++++++++++++++++++++++-- 1 file changed, 24 insertions(+), 2 deletions(-) diff --git a/README.md b/README.md index b2b09c1..1d30e6f 100644 --- a/README.md +++ b/README.md @@ -2,5 +2,27 @@ Poolformer-cnn图像融合框架 The implementation of our paper "PFCFuse: A Poolformer and CNN fusion network for Infrared-Visible Image Fusion". ## Recommended Environment: -python=3.8 -torch=1.12.1+cu113 +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 +```