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:47:40 +02:00
parent 7575ed66b1
commit 3c78c4d873
1 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
README.md
View File

@ -100,15 +100,20 @@ conda activate cddfuse
# select pytorch version yourself # select pytorch version yourself
# install cddfuse requirements # install cddfuse requirements
pip install -r requirements.txt pip install -r requirements.txt
``` ```
**2. Data Preparation** **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** **3. Pre-Processing**
Run 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** **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 ## CDDFuse