README.md 12.7 KB
Newer Older
Q
qingqing01 已提交
1

L
LielinJiang 已提交
2
English | [简体中文](./README_cn.md)
3

L
LielinJiang 已提交
4 5
# PaddleGAN

L
LielinJiang 已提交
6
PaddleGAN provides developers with high-performance implementation of classic and SOTA Generative Adversarial Networks, and supports developers to quickly build, train and deploy GANs for academic, entertainment and industrial usage.
7

L
LielinJiang 已提交
8
GAN-Generative Adversarial Network, was praised by "the Father of Convolutional Networks"  **Yann LeCun (Yang Likun)**  as **[One of the most interesting ideas in the field of computer science in the past decade]**. It's the one research area in deep learning that AI researchers are most concerned about.
L
LielinJiang 已提交
9

Q
qingqing01 已提交
10
<div align='center'>
L
LaraStuStu 已提交
11
  <img src='./docs/imgs/ppgan.jpg'>
Q
qingqing01 已提交
12
</div>
L
LielinJiang 已提交
13

L
LaraStuStu 已提交
14
[![License](https://img.shields.io/badge/license-Apache%202-red.svg)](LICENSE)![python version](https://img.shields.io/badge/python-3.6+-orange.svg)
L
LielinJiang 已提交
15

16
## 🎪 Hot Activities
L
LaraStuStu 已提交
17

18 19 20 21 22 23 24 25
- 2021.4.15~4.22

  GAN 7 Days Course Camp: Baidu Senior Research Developers help you learn the basic and advanced GAN knowledge in 7 days!

  **Courses videos and related materials: https://aistudio.baidu.com/aistudio/course/introduce/16651**

## 🚀 Recent Updates

Y
YixinKristy 已提交
26 27 28 29 30 31
- 👶 **Young or Old?:[StyleGAN V2 Face Editing](./docs/en_US/tutorials/styleganv2editing.md)-Time Machine!** 👨‍🦳
  - **[Online Toturials](https://aistudio.baidu.com/aistudio/projectdetail/3251280?channelType=0&channel=0)**
  <div align='center'>
    <img src='https://user-images.githubusercontent.com/48054808/146649047-765ec085-0a2c-4c88-9527-744836448651.gif' width='200'/>
  </div>

W
wangna11BD 已提交
32
- 🔥 **Latest Release: [PP-MSVSR](./docs/en_US/tutorials/video_super_resolution.md)** 🔥
33 34
    - **Video Super Resolution SOTA models**
  <div align='center'>
Y
YixinKristy 已提交
35
    <img src='https://user-images.githubusercontent.com/48054808/144848981-00c6ad21-0702-4381-9544-becb227ed9f0.gif' width='300'/>
36
  </div>
Y
YixinKristy 已提交
37
  
38 39 40
- 😍 **Boy or Girl?:[StyleGAN V2 Face Editing](./docs/en_US/tutorials/styleganv2editing.md)-Changing genders!** 😍
  - **[Online Toturials](https://aistudio.baidu.com/aistudio/projectdetail/2565277?contributionType=1)**
  <div align='center'>
Y
YixinKristy 已提交
41
    <img src='https://user-images.githubusercontent.com/48054808/141226707-58bd661e-2102-4fb7-8e18-c794a6b59ee8.gif' width='300'/>
42
  </div>
L
LielinJiang 已提交
43

44
- 👩‍🚀 **A Space Odyssey :[LapStyle](./docs/zh_CN/tutorials/lap_style.md) image translation take you travel around the universe**👨‍🚀
Y
YixinKristy 已提交
45 46 47 48 49 50 51 52

  - **[Online Toturials](https://aistudio.baidu.com/aistudio/projectdetail/2343740?contributionType=1)**

    <div align='center'>
      <img src='https://user-images.githubusercontent.com/48054808/133392621-9a552c46-841b-4fe4-bb24-7b0cbf86616c.gif' width='250'/>
      <img src='https://user-images.githubusercontent.com/48054808/133392630-c5329c4c-bc10-406e-a853-812a2b1f0fa6.gif' width='250'/>
      <img src='https://user-images.githubusercontent.com/48054808/133392652-f4811b1e-0676-4402-808b-a4c96c611368.gif' width='250'/>
    </div>
Y
YixinKristy 已提交
53

54
- 🧙‍♂️ **Latest Creative Project:create magic/dynamic profile for your student ID in Hogwarts** 🧙‍♀️
Y
YixinKristy 已提交
55

Y
YixinKristy 已提交
56
  - **[Online Toturials](https://aistudio.baidu.com/aistudio/projectdetail/2288888?channelType=0&channel=0)**
Y
YixinKristy 已提交
57

Y
YixinKristy 已提交
58
    <div align='center'>
Y
YixinKristy 已提交
59
      <img src='https://ai-studio-static-online.cdn.bcebos.com/da1c51844ac048aa8d4fa3151be95215eee75d8bb488409d92ec17285b227c2c' width='200'/>
Y
YixinKristy 已提交
60
    </div>
L
LielinJiang 已提交
61

62
- **💞 Add Face Morphing function💞 : you can perfectly merge any two faces and make the new face get any facial expressions!**
63 64 65 66

  - Tutorials: https://aistudio.baidu.com/aistudio/projectdetail/2254031

    <div align='center'>
Y
YixinKristy 已提交
67
      <img src='https://user-images.githubusercontent.com/48054808/128299870-66a73bb3-57a4-4985-aadc-8ddeab048145.gif' width='200'/>
68 69 70
    </div>

- **Publish a new version of First Oder Motion model by having two impressive features:**
L
LielinJiang 已提交
71
  - High resolution 512x512
72
  - Face Enhancement
L
LielinJiang 已提交
73 74
  - Tutorials: https://github.com/PaddlePaddle/PaddleGAN/blob/develop/docs/zh_CN/tutorials/motion_driving.md

Y
YixinKristy 已提交
75 76 77 78 79
- **New image translation ability--transfer photo into oil painting style:**

  - Complete tutorials for deployment: https://github.com/wzmsltw/PaintTransformer

    <div align='center'>
Y
YixinKristy 已提交
80
      <img src='https://user-images.githubusercontent.com/48054808/129904830-8b87e310-ea51-4aff-b29b-88920ee82447.png' width='500'/>
Y
YixinKristy 已提交
81
    </div>
L
LaraStuStu 已提交
82

Z
ZhengLeyizly 已提交
83 84 85
## Document Tutorial

#### **Installation**
L
LielinJiang 已提交
86

L
LielinJiang 已提交
87
* Environment dependence:
Z
ZhengLeyizly 已提交
88 89 90 91
   - PaddlePaddle >= 2.1.0
   - Python >= 3.6
   - CUDA >= 10.1
* [Full installation tutorial](https://github.com/PaddlePaddle/PaddleGAN/blob/develop/docs/zh_CN/install.md)
L
LaraStuStu 已提交
92

Z
ZhengLeyizly 已提交
93
#### **Starter Tutorial**
Q
qingqing01 已提交
94

Z
ZhengLeyizly 已提交
95 96 97 98
- [Quick start](./docs/en_US/get_started.md)
- [Data Preparation](./docs/en_US/data_prepare.md)
- [Instruction of APIs](./docs/en_US/apis/apps.md)
- [Instruction of Config Files](./docs/en_US/config_doc.md)
Q
qingqing01 已提交
99

L
LielinJiang 已提交
100
## Model Tutorial
Q
qingqing01 已提交
101

L
LielinJiang 已提交
102 103
* [Pixel2Pixel](./docs/en_US/tutorials/pix2pix_cyclegan.md)
* [CycleGAN](./docs/en_US/tutorials/pix2pix_cyclegan.md)
W
wangna11BD 已提交
104
* [LapStyle](./docs/en_US/tutorials/lap_style.md)
L
LielinJiang 已提交
105
* [PSGAN](./docs/en_US/tutorials/psgan.md)
106
* [First Order Motion Model](./docs/en_US/tutorials/motion_driving.md)
L
LielinJiang 已提交
107 108 109
* [FaceParsing](./docs/en_US/tutorials/face_parse.md)
* [AnimeGANv2](./docs/en_US/tutorials/animegan.md)
* [U-GAT-IT](./docs/en_US/tutorials/ugatit.md)
Q
qingqing01 已提交
110
* [Photo2Cartoon](./docs/en_US/tutorials/photo2cartoon.md)
L
lijianshe02 已提交
111
* [Wav2Lip](./docs/en_US/tutorials/wav2lip.md)
W
wangna11BD 已提交
112
* [Single Image Super Resolution(SISR)](./docs/en_US/tutorials/single_image_super_resolution.md)
W
wangna11BD 已提交
113
  * Including: RealSR, ESRGAN, LESRCNN, PAN, DRN
W
wangna11BD 已提交
114
* [Video Super Resolution(VSR)](./docs/en_US/tutorials/video_super_resolution.md)
W
wangna11BD 已提交
115
  * Including: ⭐ PP-MSVSR ⭐, EDVR, BasicVSR, BasicVSR++
L
LielinJiang 已提交
116
* [StyleGAN2](./docs/en_US/tutorials/styleganv2.md)
W
wangna11BD 已提交
117
* [Pixel2Style2Pixel](./docs/en_US/tutorials/pixel2style2pixel.md)
W
wangna11BD 已提交
118
* [StarGANv2](docs/en_US/tutorials/starganv2.md)
W
wangna11BD 已提交
119 120
* [MPR Net](./docs/en_US/tutorials/mpr_net.md)
* [FaceEnhancement](./docs/en_US/tutorials/face_enhancement.md)
L
LielinJiang 已提交
121

Q
qingqing01 已提交
122

L
LielinJiang 已提交
123
## Composite Application
L
LaraStuStu 已提交
124

W
wangna11BD 已提交
125
* [Video restore](./docs/en_US/tutorials/video_restore.md)
L
LaraStuStu 已提交
126

127 128 129 130 131 132
## Online Tutorial

You can run those projects in the [AI Studio](https://aistudio.baidu.com/aistudio/projectoverview/public/1?kw=paddlegan) to learn how to use the models above:

|Online Tutorial      |    link  |
|--------------|-----------|
133
|Motion Driving-multi-personal "Mai-ha-hi" | [Click and Try](https://aistudio.baidu.com/aistudio/projectdetail/1603391) |
134
|Restore the video of Beijing hundreds years ago|[Click and Try](https://aistudio.baidu.com/aistudio/projectdetail/1161285)|
135
|Motion Driving-When "Su Daqiang" sings "unravel" |[Click and Try](https://aistudio.baidu.com/aistudio/projectdetail/1048840)|
136

L
LielinJiang 已提交
137
## Examples
L
LielinJiang 已提交
138

L
LielinJiang 已提交
139
### Face Morphing
140 141 142 143 144

<div align='center'>
  <img src='https://user-images.githubusercontent.com/48054808/129020371-75de20d1-705b-44b1-8254-e09710124244.gif'width='700' />
</div>

L
LielinJiang 已提交
145
### Image Translation
L
LaraStuStu 已提交
146 147

<div align='center'>
148
  <img src='https://user-images.githubusercontent.com/48054808/119464966-d5c1c000-bd75-11eb-9696-9bb75357229f.gif'width='700' height='200'/>
L
LaraStuStu 已提交
149 150
</div>

151

L
LielinJiang 已提交
152
### Old video restore
L
LaraStuStu 已提交
153
<div align='center'>
154
  <img src='https://user-images.githubusercontent.com/48054808/119469496-fc81f580-bd79-11eb-865a-5e38482b1ae8.gif' width='700'/>
L
LaraStuStu 已提交
155 156 157
</div>


158

L
LielinJiang 已提交
159
### Motion driving
L
LaraStuStu 已提交
160
<div align='center'>
161
  <img src='https://user-images.githubusercontent.com/48054808/119469551-0a377b00-bd7a-11eb-9117-e4871c8fb9c0.gif' width='700'>
L
LaraStuStu 已提交
162 163
</div>

164

L
LielinJiang 已提交
165
### Super resolution
L
LaraStuStu 已提交
166 167

<div align='center'>
168
  <img src='https://user-images.githubusercontent.com/48054808/119469753-3e12a080-bd7a-11eb-9cde-4fa01b3201ab.png'width='700' height='250'/>
L
LaraStuStu 已提交
169 170 171
</div>


172

L
LielinJiang 已提交
173
### Makeup shifter
L
LaraStuStu 已提交
174 175

<div align='center'>
176
  <img src='https://user-images.githubusercontent.com/48054808/119469834-4ff44380-bd7a-11eb-93b6-05b705dcfbf2.png'width='700' height='250'/>
L
LielinJiang 已提交
177 178 179
</div>


180

L
LielinJiang 已提交
181
### Face cartoonization
L
LielinJiang 已提交
182 183

<div align='center'>
184
  <img src='https://user-images.githubusercontent.com/48054808/119469952-6bf7e500-bd7a-11eb-89ad-9a78b10bd4ab.png'width='700' height='250'/>
L
LaraStuStu 已提交
185 186
</div>

H
haoqiang 已提交
187

188

H
haoqiang 已提交
189 190 191
### Realistic face cartoonization

<div align='center'>
192
  <img src='https://user-images.githubusercontent.com/48054808/119470028-7f0ab500-bd7a-11eb-88e9-78a6b9e2e319.png'width='700' height='250'/>
H
haoqiang 已提交
193 194 195
</div>


196

L
LielinJiang 已提交
197
### Photo animation
Q
qingqing01 已提交
198 199

<div align='center'>
200
  <img src='https://user-images.githubusercontent.com/48054808/119470099-9184ee80-bd7a-11eb-8b12-c9400fe01266.png'width='700' height='250'/>
Q
qingqing01 已提交
201
</div>
Q
qingqing01 已提交
202

L
lijianshe02 已提交
203

204

L
lijianshe02 已提交
205 206 207
### Lip-syncing

<div align='center'>
208
  <img src='https://user-images.githubusercontent.com/48054808/119470166-a6618200-bd7a-11eb-9f98-58052ce21b14.gif'width='700'>
L
lijianshe02 已提交
209 210 211
</div>


212

L
LielinJiang 已提交
213
## Changelog
L
LielinJiang 已提交
214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233 234 235 236
- v2.1.0 (2021.12.8)
   - Release a video super-resolution model PP-MSVSR and multiple pre-training weights
   - Release several SOTA video super-resolution models and their pre-trained models such as BasicVSR, IconVSR and BasicVSR++
   - Release the light-weight motion-driven model(Volume compression: 229M->10.1M), and optimized the fusion effect
   - Release high-resolution FOMM and Wav2Lip pre-trained models
   - Release several interesting applications based on StyleGANv2, such as face inversion, face fusion and face editing
   - Released Baidu’s self-developed and effective style transfer model LapStyle and its interesting applications, and launched the official website [experience page](https://www.paddlepaddle.org.cn/paddlegan)
   - Release a light-weight image super-resolution model PAN

- v2.0.0 (2021.6.2)
  - Release [Fisrt Order Motion](https://github.com/PaddlePaddle/PaddleGAN/blob/develop/docs/en_US/tutorials/motion_driving.md) model and multiple pre-training weights
  - Release applications that support [Multi-face action driven](https://github.com/PaddlePaddle/PaddleGAN/blob/develop/docs/en_US/tutorials/motion_driving.md#1-test-for-face)
  - Release video super-resolution model [EDVR](https://github.com/PaddlePaddle/PaddleGAN/blob/develop/docs/en_US/tutorials/video_super_resolution.md) and multiple pre-training weights
  - Release the contents of [7-day punch-in training camp](https://github.com/PaddlePaddle/PaddleGAN/tree/develop/education) corresponding to PaddleGAN
  - Enhance the robustness of PaddleGAN running on the windows platform

- v2.0.0-beta (2021.3.1)
  - Completely switch the API of Paddle 2.0.0 version.
  - Release of super-resolution models: ESRGAN, RealSR, LESRCNN, DRN, etc.
  - Release lip migration model: Wav2Lip
  - Release anime model of Street View: AnimeGANv2
  - Release face animation model: U-GAT-IT, Photo2Cartoon
  - Release SOTA generation model: StyleGAN2
Q
qingqing01 已提交
237 238

- v0.1.0 (2020.11.02)
L
LielinJiang 已提交
239 240
  - Release first version, supported models include Pixel2Pixel, CycleGAN, PSGAN. Supported applications include video frame interpolation, super resolution, colorize images and videos, image animation.
  - Modular design and friendly interface.
L
LielinJiang 已提交
241

L
LielinJiang 已提交
242
## Community
L
LaraStuStu 已提交
243

L
LielinJiang 已提交
244
Scan OR Code below to join [PaddleGAN QQ Group:1058398620], you can get offical technical support  here and communicate with other developers/friends. Look forward to your participation!
L
LaraStuStu 已提交
245 246 247 248

<div align='center'>
  <img src='./docs/imgs/qq.png'width='250' height='300'/>
</div>
L
LielinJiang 已提交
249

L
LielinJiang 已提交
250
### PaddleGAN Special Interest Group(SIG)
L
LielinJiang 已提交
251

L
LielinJiang 已提交
252
It was first proposed and used by [ACM(Association for Computing Machinery)](https://en.wikipedia.org/wiki/Association_for_Computing_Machinery) in 1961. Top International open source organizations including [Kubernates](https://kubernetes.io/) all adopt the form of SIGs, so that members with the same specific interests can share, learn knowledge and develop projects. These members do not need to be in the same country/region or the same organization, as long as they are like-minded, they can all study, work, and play together with the same goals~
L
LielinJiang 已提交
253

L
LielinJiang 已提交
254
PaddleGAN SIG is such a developer organization that brings together people who interested in GAN. There are frontline developers of PaddlePaddle, senior engineers from the world's top 500, and students from top universities at home and abroad.
L
LielinJiang 已提交
255

L
LielinJiang 已提交
256
We are continuing to recruit developers interested and capable to join us building this project and explore more useful and interesting applications together.
Q
qingqing01 已提交
257

L
LielinJiang 已提交
258
SIG contributions:
Q
qingqing01 已提交
259

L
LielinJiang 已提交
260 261 262
- [zhen8838](https://github.com/zhen8838): contributed to AnimeGANv2.
- [Jay9z](https://github.com/Jay9z): contributed to DCGAN and updated install docs, etc.
- [HighCWu](https://github.com/HighCWu): contributed to c-DCGAN and WGAN. Support to use `paddle.vision.datasets`.
Q
qingqing01 已提交
263
- [hao-qiang](https://github.com/hao-qiang) & [ minivision-ai ](https://github.com/minivision-ai): contributed to the photo2cartoon project.
Q
qingqing01 已提交
264

L
LielinJiang 已提交
265

L
LielinJiang 已提交
266
## Contributing
Q
qingqing01 已提交
267

L
LielinJiang 已提交
268 269 270
Contributions and suggestions are highly welcomed. Most contributions require you to agree to a [Contributor License Agreement (CLA)](https://cla-assistant.io/PaddlePaddle/PaddleGAN) declaring.
When you submit a pull request, a CLA-bot will automatically determine whether you need to provide a CLA. Simply follow the instructions provided by the bot. You will only need to do this once across all repos using our CLA.
For more, please reference [contribution guidelines](docs/en_US/contribute.md).
L
LaraStuStu 已提交
271

L
LielinJiang 已提交
272 273
## License
PaddleGAN is released under the [Apache 2.0 license](LICENSE).