whaifree/pfcfuse
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update README.md

Browse Source
This commit is contained in:
HXY13 2024-06-03 19:48:08 +08:00 committed by GitHub
parent 054bf829e7
commit 1ad1204210
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 24 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
README.md
View File

@ -2,5 +2,27 @@
Poolformer-cnn图像融合框架 Poolformer-cnn图像融合框架
The implementation of our paper "PFCFuse: A Poolformer and CNN fusion network for Infrared-Visible Image Fusion". The implementation of our paper "PFCFuse: A Poolformer and CNN fusion network for Infrared-Visible Image Fusion".
## Recommended Environment: ## Recommended Environment:
python=3.8 python=3.8\
torch=1.12.1+cu113 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
```