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