山水画看腻了?PaddleGan带你灵魂转换!
中国山水画简称“山水画”。以山川自然景观为主要描写对象的中国画。形成于魏晋南北朝时期,但尚未从人物画中完全分离。隋唐时始独立,五代、北宋时趋于成熟,成为中国画的重要画科。传统上按画法风格分为青绿山水、金碧山水、水墨山水、浅绛山水、小青绿山水、没骨山水等。
模型效果
最左边是原图
数据出处
PaddleGan:https://github.com/PaddlePaddle/PaddleGAN/
数据集原地址:中国传统山水画数据集
论文地址:https://arxiv.org/abs/2011.05552
CycleGAN模型
CycleGAN使用未配对的图片进行图像翻译,输入具有不同样式的两个不同图像,并自动执行样式转换。CycleGAN由两个生成器和两个鉴别器组成,生成器A输入样式A的图像并输出样式B的图像,生成器B输入样式B的图像并输出样式A的图像。CycleGAN和pix2pix的最大区别在于CycleGAN可以无需在源域和目标域之间建立一一对应的映射即可实现图像转换。
准备数据
#进入PaddleGAN文件夹
%cd PaddleGAN/
# 解压数据集
!unzip -o data/data65120/painting.zip -d /home/aistudio/PaddleGAN/data
开始训练
# 训练
!python -u tools/main.py --config-file configs/painting.yaml
测试模型
# 测试
!python tools/main.py --config-file configs/painting.yaml --evaluate-only --load output_dir/epoch_20_weight.pdparams