Fix gan readme (#2565)

* Update README.md

PaddleCV/gan/README.md

@@ -6,16 +6,20 @@
 ---
 ## 内容
 
--[简介](#简介)
+- [模型简介](#模型简介)
+- [快速开始](#快速开始)
+- [进阶使用](#进阶使用)
+- [FAQ](#faq)
+- [参考论文](#参考论文)
+- [版本更新](#版本更新)
+- [作者](#作者)
 
--[快速开始](#快速开始)
-
--[参考文献](#参考文献)
-
-## 简介
+## 模型简介
 
 本图像生成模型库包含CGAN\[[3](#参考文献)\], DCGAN\[[4](#参考文献)\], Pix2Pix\[[5](#参考文献)\], CycleGAN\[[6](#参考文献)\], StarGAN\[[7](#参考文献)\], AttGAN\[[8](#参考文献)\], STGAN\[[9](#参考文献)\]。
 
+注意：AttGAN和STGAN的网络结构中，判别器去掉了instance norm。
+
 图像生成模型库库的目录结构如下：
 ```
 ├── download.py 下载数据
@@ -28,21 +32,12 @@
 │  
 ├── trainer 不同模型的训练脚本
 │   ├── CGAN.py Conditional GAN的训练脚本
-│   ├── DCGAN.py Deep Convolutional GAN的训练脚本
-│   ├── Pix2pix.py Pix2Pix GAN的训练脚本
-│   ├── CycleGAN.py CycleGAN的训练脚本
-│   ├── StarGAN.py StarGAN的训练脚本
-│   ├── AttGAN.py AttGAN的训练脚本
+│   ├── ...
 │   ├── STGAN.py STGAN的训练脚本
 │  
 ├── network 不同模型的网络结构
 │   ├── base_network.py GAN模型需要的公共基础网络结构
-│   ├── CGAN_network.py Conditional GAN的网络结构
-│   ├── DCGAN_network.py Deep Convolutional GAN的网络结构
-│   ├── Pix2pix_network.py Pix2Pix GAN的网络结构
-│   ├── CycleGAN_network.py CycleGAN的网络结构
-│   ├── StarGAN_network.py StarGAN的网络结构
-│   ├── AttGAN_network.py AttGAN的网络结构
+│   ├── ...
 │   ├── STGAN_network.py STGAN的网络结构
 │  
 ├── util 网络的基础配置和公共模块
@@ -57,10 +52,17 @@
 ```
 
 ## 快速开始
+
+### 安装说明
 **安装[PaddlePaddle](https://github.com/PaddlePaddle/Paddle)：**
 
 在当前目录下运行样例代码需要PadddlePaddle Fluid的v.1.5或以上的版本。如果你的运行环境中的PaddlePaddle低于此版本，请根据[安装文档](http://paddlepaddle.org/documentation/docs/zh/1.4/beginners_guide/install/index_cn.html)中的说明来更新PaddlePaddle。
 
+### 任务简介
+
+Pix2Pix和CycleGAN采用cityscapes\[[10](#参考文献)\]数据集进行风格转换。
+StarGAN，AttGAN和STGAN采用celeba\[[11](#参考文献)\]数据集进行属性迁移。
+
 ### 数据准备
 
 模型库中提供了download.py数据下载脚本，该脚本支持下载MNIST数据集，CycleGAN和Pix2Pix所需要的数据集。使用以下命令下载数据：
@@ -79,19 +81,26 @@ StarGAN, AttGAN和STGAN所需要的[Celeba](http://mmlab.ie.cuhk.edu.hk/projects
 
 ### 模型训练
 
-**下载预训练模型:** 
+**开始训练：** 数据准备完毕后，可以通过一下方式启动训练：
 
-本示例提供以下预训练模型:
+    python train.py \
+      --model_net=$(name_of_model) \
+      --dataset=$(name_of_dataset) \
+      --data_dir=$(path_to_data) \
+      --train_list=$(path_to_train_data_list) \
+      --test_list=$(path_to_test_data_list) \
+      --batch_size=$(batch_size)
 
-| Model| Pretrained model |
-|:---  |:---|
-| Pix2Pix  | [Pix2Pix的预训练模型]()  |
-| CycleGAN | [CycleGAN的预训练模型]() |
-| StarGAN  | [StarGAN的预训练模型]()  |
-| AttGAN   | [AttGAN的预训练模型]()   |
-| STGAN    | [STGAN的预训练模型]()     |
+- 可选参数见：
+
+    python train.py --help
+
+- 每个GAN都给出了一份运行示例，放在scripts文件夹内，用户可以直接运行训练脚本快速开始训练。
+- 用户可以通过设置`--model_net`参数来选择想要训练的模型，通过设置`--dataset`参数来选择训练所需要的数据集。
+
+### 模型测试
+模型测试是利用训练完成的生成模型进行图像生成。infer.py是主要的执行程序，调用示例如下：
 
-下载完预训练模型之后，通过设置infer.py中`--init_model`加载预训练模型，测试所需要的图片。
 执行以下命令得到CyleGAN的预测结果：
 
     python infer.py \
@@ -103,9 +112,6 @@ StarGAN, AttGAN和STGAN所需要的[Celeba](http://mmlab.ie.cuhk.edu.hk/projects
       --net_G=$(generator_network) \
       --g_base_dims=$(base_dim_of_generator)
 
-效果如图所示：
-
-
 执行以下命令得到Pix2Pix的预测结果：
 
     python infer.py \
@@ -115,51 +121,132 @@ StarGAN, AttGAN和STGAN所需要的[Celeba](http://mmlab.ie.cuhk.edu.hk/projects
       --dataset_dir=$(path_to_data) \
       --net_G=$(generator_network)
 
-效果如图所示：
-
-
-执行以下命令得到StarGAN,AttGAN和STGAN的预测结果：
+执行以下命令得到StarGAN，AttGAN或者STGAN的预测结果：
 
     python infer.py \
       --model_net=$(StarGAN_or_AttGAN_or_STGAN) \
       --init_model=$(path_to_init_model)\
       --dataset_dir=$(path_to_data)
 
-效果如图所示：
+Pix2Pix和CycleGAN的效果如图所示：
 
+<p align="centor">
+    <img src = "images/pix2pix_cyclegan.png" width=550><br/>
+    Pix2Pix和CycleGAN的效果图
+</p>
 
-**开始训练：** 数据准备完毕后，可以通过一下方式启动训练：
 
-    python train.py \
-      --model_net=$(name_of_model) \
-      --dataset=$(name_of_dataset) \
-      --data_dir=$(path_to_data) \
-      --train_list=$(path_to_train_data_list) \
-      --test_list=$(path_to_test_data_list) \
-      --batch_size=$(batch_size)
+StarGAN，AttGAN和STGAN的效果如图所示：
 
-- 可选参数见：
+<p align="centor">
+    <img src = "images/female_stargan_attgan_stgan.png" width=550><br/>
+    StarGAN,AttGAN和STGAN的效果图
+</p>
 
-    python train.py --help
-    
-- 每个GAN都给出了一份运行示例，放在scripts文件夹内，用户可以直接运行训练脚本快速开始训练。
-- 用户可以通过设置model_net参数来选择想要训练的模型，通过设置dataset参数来选择训练所需要的数据集。
 
-### 模型测试
-模型测试是利用训练完成的生成模型进行图像生成。infer.py是主要的执行程序，调用示例如下：
 
-    python infer.py \
-      --model_net=$(name_of_model) \
-      --init_model=$(path_to_model) \
-      --dataset_dir=$(path_to_data)
-      
 - 每个GAN都给出了一份测试示例，放在scripts文件夹内，用户可以直接运行测试脚本得到测试结果。
 
+**下载预训练模型:**
+
+本示例提供以下预训练模型:
+
+| Model| Pretrained model |
+|:---|:---|
+| Pix2Pix  | [Pix2Pix的预训练模型](https://paddle-gan-models.bj.bcebos.com/pix2pix_G.tar.gz)  |
+| CycleGAN | [CycleGAN的预训练模型](https://paddle-gan-models.bj.bcebos.com/cyclegan_9blocks_G.tar.gz) |
+| StarGAN  | [StarGAN的预训练模型](https://paddle-gan-models.bj.bcebos.com/stargan_G.tar.gz)  |
+| AttGAN   | [AttGAN的预训练模型]()   |
+| STGAN    | [STGAN的预训练模型](https://paddle-gan-models.bj.bcebos.com/stgan_G.tar.gz)    |
+
+
+## 进阶使用
+### 背景介绍
+
+CGAN，条件生成对抗网络，一种带条件约束的GAN，使用额外信息对模型增加条件，可以指导数据生成过程。
+
+DCGAN，深度卷积生成对抗网络，将GAN和卷积网络结合起来，以解决GAN训练不稳定的问题，利用卷积神经网络作为网络结构进行图像生成，可以得到更加丰富的层次表达。
+
+Pix2Pix利用成对的图片进行图像翻译，即输入为同一张图片的两种不同风格，可用于进行风格迁移。
+
+CycleGAN可以利用非成对的图片进行图像翻译，即输入为两种不同风格的不同图片，自动进行风格转换。
+
+StarGAN多领域属性迁移，引入辅助分类帮助单个判别器判断多个属性，可用于人脸属性转换。
+
+AttGAN利用分类损失和重构损失来保证改变特定的属性，可用于人脸特定属性转换。
+
+STGAN只输入有变化的标签，引入GRU结构，更好的选择变化的属性，可用于人脸特定属性转换。
+
+### 模型概览
+
+- Pix2Pix由一个生成网络和一个判别网络组成。生成网络中编码部分的网络结构都是采用`convolution-batch norm-ReLU`作为基础结构，解码部分的网络结构由`transpose convolution-batch norm-ReLU`组成，判别网络基本是由`convolution-norm-leaky_ReLU`作为基础结构，详细的网络结构可以查看`network/Pix2pix_network.py`文件。生成网络提供两种可选的网络结构：Unet网络结构和普通的encoder-decoder网络结构。网络利用损失函数学习从输入图像到输出图像的映射，生成网络损失函数由CGAN的损失函数和L1损失函数组成，判别网络损失函数由CGAN的损失函数组成。生成器的网络结构如下图所示：
+
+<p align="centor">
+    <img src = "images/pix2pix_gen.png" width=550><br/>
+    Pix2Pix生成网络结构图[5]
+</p>
+
+
+- CycleGAN由两个生成网络和两个判别网络组成，生成网络A是输入A类风格的图片输出B类风格的图片，生成网络B是输入B类风格的图片输出A类风格的图片。生成网络中编码部分的网络结构都是采用`convolution-norm-ReLU`作为基础结构，解码部分的网络结构由`transpose convolution-norm-ReLU`组成，判别网络基本是由`convolution-norm-leaky_ReLU`作为基础结构，详细的网络结构可以查看`network/CycleGAN_network.py`文件。生成网络提供两种可选的网络结构：Unet网络结构和普通的encoder-decoder网络结构。生成网络损失函数由CGAN的损失函数，重构损失和自身损失组成，判别网络的损失函数由CGAN的损失函数组成。
+
+<p align="centor">
+    <img src = "images/pix2pix_gen.png" width=550><br/>
+    CycleGAN生成网络结构图[5]
+</p>
+
+
+- StarGAN中生成网络的编码部分主要由`convolution-instance norm-ReLU`组成，解码部分主要由`transpose convolution-norm-ReLU`组成，判别网络主要由`convolution-leaky_ReLU`组成，详细网络结构可以查看`network/StarGAN_network.py`文件。生成网络的损失函数是由CGAN的损失函数，重构损失和分类损失组成，判别网络的损失函数由预测损失，分类损失和梯度惩罚损失组成。
+
+<p align="centor">
+    <img src = "images/stargan_gen.png" width=300><br/>
+    StarGAN生成网络结构[7]
+</p>
+
+<p align="centor">
+    <img src = "images/stargan_dis.png" width=300><br/>
+    StarGAN判别网络结构[7]
+</p>
+
+
 
-## 参考文献
+- AttGAN中生成网络的编码部分主要由`convolution-instance norm-ReLU`组成，解码部分由`transpose convolution-norm-ReLU`组成，判别网络主要由`convolution-leaky_ReLU`组成，详细网络结构可以查看`network/AttGAN_network.py`文件。生成网络的损失函数是由CGAN的损失函数，重构损失和分类损失组成，判别网络的损失函数由预测损失，分类损失和梯度惩罚损失组成。
+
+<p align="centor">
+    <img src = "images/attgan_net.png" width=800><br/>
+    AttGAN的网络结构[8]
+</p>
+
+
+- STGAN中生成网络再编码器和解码器之间加入Selective Transfer Units\(STU\)，有选择的转换编码网络，从而更好的适配解码网络。生成网络中的编码网络主要由`convolution-instance norm-ReLU`组成，解码网络主要由`transpose convolution-norm-leaky_ReLU`组成，判别网络主要由`convolution-leaky_ReLU`组成，详细网络结构可以查看`network/STGAN_network.py`文件。生成网络的损失函数是由CGAN的损失函数，重构损失和分类损失组成，判别网络的损失函数由预测损失，分类损失和梯度惩罚损失组成。
+
+<p align="centor">
+    <img src = "images/stgan_net.png" width=800><br/>
+    STGAN的网络结构[9]
+</p>
+
+
+注意：网络结构中的norm指的是用户可以选用batch norm或者instance norm来搭建自己的网络。
+
+
+## FAQ
+
+**Q：** StarGAN/AttGAN/STGAN中属性没有变化，为什么？
+
+**A：** 查看是否所有的标签都转换对了。
+
+**Q：** 预测结果不正常，是怎么回事？
+
+**A：** 某些GAN预测的时候batch_norm的设置需要和训练的时候行为一致，查看模型库中相应的GAN中预测时batch_norm的行为和自己模型中的预测时batch_norm的行为是否一致。
+
+**Q：** 为什么STGAN和ATTGAN中变男性得到的预测结果是变女性呢？
+
+**A：** 这是由于预测时标签的设置，目标标签是基于原本的标签进行改变，比如原本图片是男生，预测代码对标签进行转变的时候会自动变成相对立的标签，即女性，所以得到的结果是女生。如果想要原本是男生，转变之后还是男生，可以参考模型库中预测代码的StarGAN的标签设置。
+
+
+## 参考论文
 [1] [Goodfellow, Ian J.; Pouget-Abadie, Jean; Mirza, Mehdi; Xu, Bing; Warde-Farley, David; Ozair, Sherjil; Courville, Aaron; Bengio, Yoshua. Generative Adversarial Networks. 2014. arXiv:1406.2661 [stat.ML].](https://arxiv.org/abs/1406.2661)
 
-[2] [https://zh.wikipedia.org/wiki/生成对抗网络](https://zh.wikipedia.org/wiki/生成对抗网络)
+[2] [生成对抗网络](https://zh.wikipedia.org/wiki/生成对抗网络)
 
 [3] [Conditional Generative Adversarial Nets](https://arxiv.org/abs/1411.1784)
 
@@ -174,3 +261,20 @@ StarGAN, AttGAN和STGAN所需要的[Celeba](http://mmlab.ie.cuhk.edu.hk/projects
 [8] [AttGAN: Facial Attribute Editing by Only Changing What You Want](https://arxiv.org/abs/1711.10678)
 
 [9] [STGAN: A Unified Selective Transfer Network for Arbitrary Image Attribute Editing](https://arxiv.org/abs/1904.09709)
+
+[10] [The Cityscapes Dataset for Semantic Urban Scene Understanding](https://arxiv.org/abs/1604.01685)
+
+[11] [Deep Learning Face Attributes in the Wild](https://arxiv.org/abs/1411.7766)
+
+
+## 版本更新
+
+- 4/2019 新增CGAN, DCGAN, Pix2Pix, CycleGAN
+- 6/2019 新增StarGAN, AttGAN, STGAN
+
+## 作者
+- [ceci3](https://github.com/ceci3)
+- [zhumanyu](https://github.com/zhumanyu)
+
+## 如何贡献代码
+如果你可以修复某个issue或者增加一个新功能，欢迎给我们提交PR。如果对应的PR被接受了，我们将根据贡献的质量和难度进行打分（0-5分，越高越好）。如果你累计获得了10分，可以联系我们获得面试机会或者为你写推荐信。