Update README.md
This commit is contained in:
parent
054bf829e7
commit
1ad1204210
26
README.md
26
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
|
||||
```
|
||||
|
Loading…
Reference in New Issue
Block a user