From 069ee192085534e070d552d627dcba231de15509 Mon Sep 17 00:00:00 2001
From: LielinJiang <50691816+LielinJiang@users.noreply.github.com>
Date: Tue, 9 Feb 2021 10:59:30 +0800
Subject: [PATCH] update docs configs (#161)

---
 README.md                                |  2 ++
 README_cn.md                             |  2 ++
 configs/esrgan_psnr_x4_div2k.yaml        |  4 ++--
 docs/en_US/tutorials/super_resolution.md | 20 +++++++++++++++++++-
 4 files changed, 25 insertions(+), 3 deletions(-)

diff --git a/README.md b/README.md
index 79b751d..cd5008a 100644
--- a/README.md
+++ b/README.md
@@ -45,6 +45,8 @@ GAN-Generative Adversarial Network, was praised by "the Father of Convolutional
 * [Photo2Cartoon](./docs/en_US/tutorials/photo2cartoon.md)
 * [Wav2Lip](./docs/en_US/tutorials/wav2lip.md)
 * [Super_Resolution](./docs/en_US/tutorials/super_resolution.md)
+* [StyleGAN2](./docs/en_US/tutorials/styleganv2.md)
+
 
 ## Composite Application
 
diff --git a/README_cn.md b/README_cn.md
index ded54c9..f0b8414 100644
--- a/README_cn.md
+++ b/README_cn.md
@@ -46,6 +46,8 @@ GAN--生成对抗网络，被“卷积网络之父”**Yann LeCun（杨立昆）
 * [Photo2Cartoon](docs/zh_CN/tutorials/photo2cartoon.md)
 * [Wav2Lip](docs/zh_CN/tutorials/wav2lip.md)
 * [Super_Resolution](./docs/en_US/tutorials/super_resolution.md)
+* [StyleGAN2](./docs/en_US/tutorials/styleganv2.md)
+
 
 ## 复合应用
 
diff --git a/configs/esrgan_psnr_x4_div2k.yaml b/configs/esrgan_psnr_x4_div2k.yaml
index 23705d7..5df56c9 100644
--- a/configs/esrgan_psnr_x4_div2k.yaml
+++ b/configs/esrgan_psnr_x4_div2k.yaml
@@ -18,8 +18,8 @@ model:
 dataset:
   train:
     name: SRDataset
-    gt_folder: data/DIV2K/DIV2K_train_HR
-    lq_folder: data/DIV2K/DIV2K_train_LR_bicubic/X4
+    gt_folder: data/DIV2K/DIV2K_train_HR_sub
+    lq_folder: data/DIV2K/DIV2K_train_LR_bicubic/X4_sub
     num_workers: 4
     batch_size: 16
     scale: 4
diff --git a/docs/en_US/tutorials/super_resolution.md b/docs/en_US/tutorials/super_resolution.md
index c5f0d1b..b44469c 100644
--- a/docs/en_US/tutorials/super_resolution.md
+++ b/docs/en_US/tutorials/super_resolution.md
@@ -51,8 +51,26 @@
   ```
     python data/process_div2k_data.py --data-root data/DIV2K
   ```
+  When the program is finished, check whether there are ``DIV2K_train_HR_sub``, ``X2_sub``, ``X3_sub`` and ``X4_sub`` directories in the DIV2K directory
+  ```
+    PaddleGAN
+      ├── data
+          ├── DIV2K
+                ├── DIV2K_train_HR
+                ├── DIV2K_train_HR_sub
+                ├── DIV2K_train_LR_bicubic
+                |    ├──X2
+                |    ├──X2_sub
+                |    ├──X3
+                |    ├──X3_sub
+                |    ├──sX4
+                |    └──X4_sub
+                ├── DIV2K_valid_HR
+                ├── DIV2K_valid_LR_bicubic
+              ...
+  ```
 
-  Prepare dataset for realsr df2k model:
+#### Prepare dataset for realsr df2k model
   Download dataset from [NTIRE 2020 RWSR](https://competitions.codalab.org/competitions/22220#participate) and unzip it to your path.
   Unzip Corrupted-tr-x.zip and Corrupted-tr-y.zip to ``PaddleGAN/data/ntire20`` directory.
 
-- 
GitLab