fanwuyang/Stripformer
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update README.md

Browse Source
This commit is contained in:
pp00704831
2022-07-16 22:00:17 +08:00
committed by GitHub
parent a5a6d842b7
commit 8aebc0a622
1 changed files with 16 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files
README.md
+16 -12
View File
@@ -22,12 +22,14 @@ Download "[GoPro](https://drive.google.com/drive/folders/1BdV2l7A5MRXLWszGonMxR8
For example: './datasets/GoPro/train/blur/\*\*/\*.png' For example: './datasets/GoPro/train/blur/\*\*/\*.png'
**We train our Stripformer in two stages:** </br> **We train our Stripformer in two stages:** </br>
**1) We pre-train Stripformer for 3000 epochs on patch size 256x256. Please run the following commands.** </br> * We pre-train Stripformer for 3000 epochs with patch size 256x256 </br>
* Run the following commands
``` ```
python train_Stripformer_pretrained.py python train_Stripformer_pretrained.py
``` ```
**2) After stage 1, we keep training Stripformer for 1000 epochs on patch size 512x512. Please run the following commands.** </br> * After 3000 epochs, we keep training Stripformer for 1000 epochs with patch size 512x512 </br>
* Run the following commands
``` ```
python train_Stripformer_gopro.py python train_Stripformer_gopro.py
``` ```
@@ -37,19 +39,21 @@ For reproducing our results on GoPro and HIDE dataset, download the "[Stripforme
For reproducing our results on RealBlur dataset, download "[Stripformer_realblur_J.pth](https://drive.google.com/drive/folders/1YcIwqlgWQw_dhy_h0fqZlnKGptq1eVjZ?usp=sharing)" and "[Stripformer_realblur_R.pth](https://drive.google.com/drive/folders/1YcIwqlgWQw_dhy_h0fqZlnKGptq1eVjZ?usp=sharing)" For reproducing our results on RealBlur dataset, download "[Stripformer_realblur_J.pth](https://drive.google.com/drive/folders/1YcIwqlgWQw_dhy_h0fqZlnKGptq1eVjZ?usp=sharing)" and "[Stripformer_realblur_R.pth](https://drive.google.com/drive/folders/1YcIwqlgWQw_dhy_h0fqZlnKGptq1eVjZ?usp=sharing)"
* For testing on GoPro test set </br> **For testing on GoPro dataset** </br>
Download "[GoPro](https://drive.google.com/drive/folders/1BdV2l7A5MRXLWszGonMxR88eV27geb_n?usp=sharing)" full dataset or test set into './datasets' </br> * Download "[GoPro](https://drive.google.com/drive/folders/1BdV2l7A5MRXLWszGonMxR88eV27geb_n?usp=sharing)" full dataset or test set into './datasets' (For example: './datasets/GoPro/test/blur/\*\*/\*.png') </br>
For example: './datasets/GoPro/test/blur/\*\*/\*.png' * Run the following commands
``` ```
python predict_GoPro_test_results.py --weights_path ./Stripformer_gopro.pth python predict_GoPro_test_results.py --weights_path ./Stripformer_gopro.pth
``` ```
* For testing on HIDE dataset </br> **For testing on HIDE dataset** </br>
Download "[HIDE](https://drive.google.com/drive/folders/1BdV2l7A5MRXLWszGonMxR88eV27geb_n?usp=sharing)" into './datasets' </br> * Download "[HIDE](https://drive.google.com/drive/folders/1BdV2l7A5MRXLWszGonMxR88eV27geb_n?usp=sharing)" into './datasets' </br>
* Run the following commands
``` ```
python predict_HIDE_results.py --weights_path ./Stripformer_gopro.pth python predict_HIDE_results.py --weights_path ./Stripformer_gopro.pth
``` ```
* For testing on RealBlur test sets **For testing on RealBlur test sets** </br>
Download "[RealBlur_J](https://drive.google.com/drive/folders/1BdV2l7A5MRXLWszGonMxR88eV27geb_n?usp=sharing)" and "[RealBlur_R](https://drive.google.com/drive/folders/1BdV2l7A5MRXLWszGonMxR88eV27geb_n?usp=sharing)" into './datasets' </br> * Download "[RealBlur_J](https://drive.google.com/drive/folders/1BdV2l7A5MRXLWszGonMxR88eV27geb_n?usp=sharing)" and "[RealBlur_R](https://drive.google.com/drive/folders/1BdV2l7A5MRXLWszGonMxR88eV27geb_n?usp=sharing)" into './datasets' </br>
* Run the following commands
``` ```
python predict_RealBlur_J_test_results.py --weights_path ./Stripformer_realblur_J.pth python predict_RealBlur_J_test_results.py --weights_path ./Stripformer_realblur_J.pth
``` ```
@@ -57,9 +61,9 @@ python predict_RealBlur_J_test_results.py --weights_path ./Stripformer_realblur_
python predict_RealBlur_R_test_results.py --weights_path ./Stripformer_realblur_R.pth python predict_RealBlur_R_test_results.py --weights_path ./Stripformer_realblur_R.pth
``` ```
* For testing your own training weight (take GoPro for a example) **For testing your own training weight (take GoPro for a example)** </br>
1) Rename the path in line 23 in the predict_GoPro_test_results.py </br> * Rename the path in line 23 in the predict_GoPro_test_results.py </br>
2) Chage command to --weights_path ./final_Stripformer_gopro.pth * Chage command to --weights_path ./final_Stripformer_gopro.pth
## Evaluation ## Evaluation
* For evaluation on GoPro results in MATLAB, download "[Stripformer_GoPro_results](https://drive.google.com/drive/folders/19uXbEEHojEwC29_jL8Gkd1jknc8kiRcR?usp=sharing)" into './out' * For evaluation on GoPro results in MATLAB, download "[Stripformer_GoPro_results](https://drive.google.com/drive/folders/19uXbEEHojEwC29_jL8Gkd1jknc8kiRcR?usp=sharing)" into './out'