From 5a6340a6effe5d6b35ba6523d465003110540fcc Mon Sep 17 00:00:00 2001 From: pp00704831 <57279608+pp00704831@users.noreply.github.com> Date: Sat, 16 Jul 2022 23:36:00 +0800 Subject: [PATCH] Update README.md --- README.md | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/README.md b/README.md index 08224bb..ff237a5 100644 --- a/README.md +++ b/README.md @@ -23,13 +23,13 @@ For example: './datasets/GoPro' **We train our Stripformer in two stages:**
* We pre-train Stripformer for 3000 epochs with patch size 256x256
-* Run the following commands +* Run the following command ``` python train_Stripformer_pretrained.py ``` * After 3000 epochs, we keep training Stripformer for 1000 epochs with patch size 512x512
-* Run the following commands +* Run the following command ``` python train_Stripformer_gopro.py ``` @@ -41,19 +41,19 @@ For reproducing our results on RealBlur dataset, download "[Stripformer_realblur **For testing on GoPro dataset**
* Download "[GoPro](https://drive.google.com/drive/folders/1BdV2l7A5MRXLWszGonMxR88eV27geb_n?usp=sharing)" full dataset or test set into './datasets' (For example: './datasets/GoPro/test')
-* Run the following commands +* Run the following command ``` python predict_GoPro_test_results.py --weights_path ./Stripformer_gopro.pth ``` **For testing on HIDE dataset**
* Download "[HIDE](https://drive.google.com/drive/folders/1BdV2l7A5MRXLWszGonMxR88eV27geb_n?usp=sharing)" into './datasets'
-* Run the following commands +* Run the following command ``` python predict_HIDE_results.py --weights_path ./Stripformer_gopro.pth ``` **For testing on RealBlur test sets**
* 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'
-* Run the following commands +* Run the following command ``` python predict_RealBlur_J_test_results.py --weights_path ./Stripformer_realblur_J.pth ```