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

Update README.md

Browse Source
This commit is contained in:
Zhaozixiang1228 2023-06-17 00:51:09 +02:00
parent b4a336be9b
commit 9ea7476527
1 changed files with 10 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
README.md
View File

@ -104,16 +104,23 @@ pip install -r requirements.txt
**2. Data Preparation**
Download the MSRS dataset from [this link](https://github.com/Linfeng-Tang/MSRS) and place it in the folder ``'./MSRS_train'``.
Download the MSRS dataset from [this link](https://github.com/Linfeng-Tang/MSRS) and place it in the folder ``'./MSRS_train/'``.
**3. Pre-Processing**
Run
```python dataprocessing.py``` and the processed training dataset is in ``'./data/MSRS_train_imgsize_128_stride_200.h5'``.
```
python dataprocessing.py
```
and the processed training dataset is in ``'./data/MSRS_train_imgsize_128_stride_200.h5'``.
**4. CDDFuse Training**
Run ```python train.py``` and the trained model is available in ``'./models/'``.
Run
```
python train.py
```
and the trained model is available in ``'./models/'``.
## CDDFuse