图片视频修复-baidu paddleGAN应用小样

1、概述

paddlepaddle（百度飞浆）作为百度的人工智能框架，在发展上可能没有tensorflow与pytorch来的成熟。但在本土化的应用中也有非常的明显的优势，比如汉语的自然语言处理层面。为了保证平台能够快速上手。他也内置了许多有趣的功能模块。本文主要介绍paddleGAN，对抗生成模块的相关接口。可以帮助我们达到一些意想不到的效果。下面的内容将涵盖图片的上色，视频的修复，超频以及卡通头像生成等内容。

在运行本文的案例之前您需要先安装paddlepaddle以及panddleGAN具体的安装步骤请参考下面链接。

https://github.com/PaddlePaddle/PaddleGAN/blob/develop/docs/zh_CN/install.md

2、引入相关类库并设置GPU运行环境

from matplotlib import pyplot as plt
%matplotlib notebook
from PIL import Image
import cv2
import imageio
import numpy as np
import matplotlib.pyplot as plt
import matplotlib.animation as animation
from IPython.display import HTML
import warnings
warnings.filterwarnings("ignore")

import paddle
paddle.set_device('gpu')

如果您是单卡GPU可以看到下面输出

CUDAPlace(0)

编写显示视频辅助函数

# 定义一个展示视频的函数
def display(driving, fps, size=(8, 6)):
    fig = plt.figure(figsize=size)

    ims = []
    for i in range(len(driving)):
        cols = []
        cols.append(driving[i])

        im = plt.imshow(np.concat