PaddleGAN带你沉浸式漫游太空 (ﾉ･ｪ･)ﾉ：实现高质量风格迁移

【PaddleGAN沉浸式漫游太空👨‍🚀】教你快速实现高质量的艺术风格迁移

无需宇宙飞船也可以漫游宇宙？！👨‍🚀👩‍🚀

你不是在做梦！🚀

AI艺术风格迁移让你美梦成真！！！🛰

不用觉得遥不可及，今天PaddleGAN就用AI的力量带领大家踏上5大行星奇幻之旅，让你“沉浸式”体验奇幻的宇宙!

它能把行星的幻妙与我们周边的草长鸢飞结合！让你在感受日常自然风光、城市光景的同时，不可预料的经历一系列奇幻趣景（友情提示大家提前买好“惊奇险”、“惊吓险”、“震惊险”、“过嗨险”等相关保险╮(╯▽╰)╭ ）

好的，万事俱备，PaddleGAN号宇宙飞船即将起飞，还未登机的开发者们抓紧扫码上车！

LapStyle 原理剖析

LapStyle，由百度视觉团队所研发的一种基于拉普拉斯金字塔的高质量风格化方法，不仅兼顾风格迁移的质量和速度，还能灵活训练各类风格！ 可谓是风格迁移领域的佼佼者！

有多厉害呢？其论文 ”Drafting and Revision: Laplacian Pyramid Network for Fast High-Quality Artistic Style Transfer” 已被CVPR2021收录！

最最最良心的是，其相关代码已经开源于飞桨生成对抗网络开发套件PaddleGAN中，并且提供四种预训练风格模型，一行代码直接推理！分分钟GET国际名画同款风格！！

下载PaddleGAN & 安装相关依赖

# 从github上克隆PaddleGAN代码（如下载速度过慢，可用gitee源）
#!git clone https://github.com/PaddlePaddle/PaddleGAN
#!git clone https://gitee.com/PaddlePaddle/PaddleGAN

# 安装依赖
%cd /home/aistudio/PaddleGAN
!pip install -r requirements.txt

%cd PaddleGAN
!git checkout develop

使用预训练模型

PaddleGAN官网中内置了四种风格，运行一行代码即可实现风格迁移！

对应以下：StarryNew、Stars、Ocean、Circuit

除此之外，本项目帮大家训练好了五大行星+粉色星球的风格模型，大家可以一键推理使用

预训练模型路径：/home/aistudio/星球风格预训练模型

• --content_img (str): 输入的内容图像路径。
• --output_path (str): 输出的图像路径，默认为output_dir。
• --weight_path (str): 模型权重路径，设置None时会自行下载预训练模型，默认为None。
• --style (str): 生成图像风格，当weight_path为None时，可以在starrynew, circuit, ocean 和 stars中选择，默认为starrynew。
• --style_image_path (str): 输入的风格图像路径，当weight_path不为None时需要输入，默认为None。

大家只需替换 --content_img (str) 为自己想要做风格化的原图、--output_path (str)为结果输出的路径、--weight_path (str)为想要的风格的预训练参数、--style_image_path (str)为想要的风格的星球图片。

!python applications/tools/lapstyle.py  \
        --content_img '/home/aistudio/地球上的风景&建筑/玉兔.png' \
        --output_path '/home/aistudio/月球风光/玉兔' \
        --weight_path '/home/aistudio/星球风格预训练模型/moon_iter_30000_weight.pdparams' \
        --style_image_path '/home/aistudio/星球图片/月球.png'

训练自定义风格

如果不满足于以上四种风格，PaddleGAN还支持大家自己训练自定义风格，只需下载训练数据集coco2017，再替换自己想要的特定风格即可~

一共需要训练三个模型：lapstyle_draft, lapstyle_rev_first, lapstyle_rev_second

按照顺序依次训练，上一个训练的模型参数作为下一个模型的输入，直到完成三个模型的训练得到最后的模型训练参数即可~

• 友情提示：每次训练大概需要4个小时左右，完成一次风格模型的全部训练，大概需要12小时（GPU）

#解压数据集
!unzip -oq /home/aistudio/data/data97273/annotations_trainval2017.zip
!unzip -oq /home/aistudio/data/data97273/train2017.zip
!unzip /home/aistudio/data/data7122/test2017.zip

#训练第一个模型
!python -u tools/main.py --config-file configs/lapstyle_draft.yaml 

#训练第二个模型
!python -u tools/main.py --config-file configs/lapstyle_rev_first.yaml --resume output_dir/lapstyle_rev_first-2021-09-09-18-02/iter_20000_checkpoint.pdparams

#训练第三个模型
!python -u tools/main.py --config-file configs/lapstyle_rev_second.yaml --resume output_dir/lapstyle_rev_second-2021-09-10-12-38/iter_25000_checkpoint.pdparams

结果展示

PaddleGAN带你沉浸式漫游太空 (ﾉ･ｪ･)ﾉ：实现高质量风格迁移