From 4049cffab758e37ef4455dec5eb7d1cc25edfe77 Mon Sep 17 00:00:00 2001 From: pp00704831 <57279608+pp00704831@users.noreply.github.com> Date: Sat, 16 Jul 2022 23:34:53 +0800 Subject: [PATCH] Update README.md --- README.md | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/README.md b/README.md index 817baca..08224bb 100644 --- a/README.md +++ b/README.md @@ -19,7 +19,7 @@ pip install albumentations==1.1.0 ## Training Download "[GoPro](https://drive.google.com/drive/folders/1BdV2l7A5MRXLWszGonMxR88eV27geb_n?usp=sharing)" dataset into './datasets'
-For example: './datasets/GoPro/train/blur/\*\*/\*.png' +For example: './datasets/GoPro' **We train our Stripformer in two stages:**
* We pre-train Stripformer for 3000 epochs with patch size 256x256
@@ -40,7 +40,7 @@ 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 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/blur/\*\*/\*.png')
+* 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 ``` python predict_GoPro_test_results.py --weights_path ./Stripformer_gopro.pth