Python PaddleGAN实现调整照片人物年龄

 更新时间:2021年12月30日 08:52:44   作者:剑客阿良_ALiang  
这篇文章主要介绍了通过PaddleGAN实现照片人物的老年化和年轻化处理,文中的示例代码讲解有效,对我们学习或工作有一定的帮助,感兴趣的可以学习一下

前言

最近在试着研究飞浆平台的许多功能,看到了许多有意思的功能。其中可以将照片美化以及年龄调整这个功能让我想到了之前抖音的一个功能,所以特别感兴趣。花了些时间把项目拉下来玩了玩,用了一些我自己找的数据。

PaddleGAN的Github地址:github仓库

环境部署

如果没有看过相关的文章,可能会被README搞得很迷糊。先不用看README中一个个教程或者md,我们要先安装执行环境。主要看docs/zh_CN/install.md文件。

先看一下我们需要安装什么。项目已经提供了paddlepaddle的安装命令,按照自己机器的情况选择CUDA或者CPU。

还有一些包需要补充安装的。

注意哦,还需要安装dlib,按照我下面的命令执行一下。

pip install dlib -i https://pypi.douban.com/simple

OK,我们环境安装好了,我们开始玩玩代码吧。

项目使用

因为PaddleGAN实现照片编辑功能核心是StyleGAN,大家有兴趣的话也可以参看一下StyleGAN项目。在编辑图片前,需要先提取图片的Latant Code(潜在因子)。简而言之,会先将图片做预处理,然后使用预处理的数据进行编辑效果。

预处理部分

按照这个思路,我们先找到预处理md说明文件。看看下面的内容。

我们看看pixel2style2pixel.py的参数说明。

可以按照你的需求修改一下项目给出的命令。我这边使用的参数都是项目给出的,主要测试一下效果。从参数说明我们可以看出我们最重要需要提供的两个参数,分别是图片路径和输出目录。

先放出我准备的照片,大家都非常熟悉。

执行命令

python -u applications/tools/pixel2style2pixel.py --input_image D:/spyder/PaddleGAN/data_hy/337.jpg --output_path  D:/spyder/PaddleGAN/data_hy/output --model_type ffhq-inversion --seed 233 --size 1024 --style_dim 512 --n_mlp 8 --channel_multiplier 2 --cpu

执行效果(第一次会下载模型)

我们可以在输出目录看到3个文件,其中dst.npy就是后面我们需要做效果的npy数据。我们先看看预处理后的图片什么样。

可以看出细节变清晰了,照片集中在头部位置,头发也做了一定的处理,无官稍微修整了一下。

照片老化处理

下面我们继续看一下另一个md文件。

我们看到另一个py文件可以帮助我们实现图片的编辑,styleganv2editing.py。我们先看看参数说明。

注意哦,不只是可以变换年龄,还有其他的选项。age、eyes_open、eye_distance、eye_eyebrow_distance、eye_ratio、gender、lip_ratio、mouth_open、mouth_ratio、nose_mouth_distance、nose_ratio、nose_tip、pitch、roll、smile、yaw。那么其实还有很多效果我们可以实现。

下面是我准备的老化图片命令

python -u applications/tools/styleganv2editing.py --latent D:/spyder/PaddleGAN/data_hy/output/dst.npy --output_path D:/spyder/PaddleGAN/data_hy/old --model_type ffhq-config-f --size 1024 --style_dim 512 --n_mlp 8 --channel_multiplier 2 --direction_name age --direction_offset 3 --cpu

执行情况

(pytorch) D:\spyder\PaddleGAN>python -u applications/tools/styleganv2editing.py --latent D:/spyder/PaddleGAN/data_hy/output/dst.npy --output_path D:/spyder/PaddleGAN/data_hy/old --model
_type ffhq-config-f --size 1024 --style_dim 512 --n_mlp 8 --channel_multiplier 2 --direction_name age --direction_offset 3 --cpu
C:\ProgramData\Anaconda3\envs\pytorch\lib\site-packages\skimage\data\__init__.py:107: DeprecationWarning:
    Importing file_hash from pooch.utils is DEPRECATED. Please import from the
    top-level namespace (`from pooch import file_hash`) instead, which is fully
    backwards compatible with pooch >= 0.1.
 
  return file_hash(path) == expected_hash
[12/29 20:17:06] ppgan INFO: Found C:\Users\huyi\.cache\ppgan\stylegan2-ffhq-config-f.pdparams
[12/29 20:17:09] ppgan INFO: Found C:\Users\huyi\.cache\ppgan\stylegan2-ffhq-config-f-directions.pdparams

看看效果

老态还是很明显的,效果不错。

照片年轻化处理

和上面一样,下面是我的年轻化命令:

python -u applications/tools/styleganv2editing.py --latent D:/spyder/PaddleGAN/data_hy/output/dst.npy --output_path D:/spyder/PaddleGAN/data_hy/young --model_type ffhq-config-f --size 1024 --style_dim 512 --n_mlp 8 --channel_multiplier 2 --direction_name age --direction_offset -3 --cpu

对比两个命令可以看出主要是--direction_offset参数不同。

执行情况

(pytorch) D:\spyder\PaddleGAN>python -u applications/tools/styleganv2editing.py --latent D:/spyder/PaddleGAN/data_hy/output/dst.npy --output_path D:/spyder/PaddleGAN/data_hy/young --mod
el_type ffhq-config-f --size 1024 --style_dim 512 --n_mlp 8 --channel_multiplier 2 --direction_name age --direction_offset -3 --cpu
C:\ProgramData\Anaconda3\envs\pytorch\lib\site-packages\skimage\data\__init__.py:107: DeprecationWarning:
    Importing file_hash from pooch.utils is DEPRECATED. Please import from the
    top-level namespace (`from pooch import file_hash`) instead, which is fully
    backwards compatible with pooch >= 0.1.
 
  return file_hash(path) == expected_hash
[12/29 20:20:07] ppgan INFO: Found C:\Users\huyi\.cache\ppgan\stylegan2-ffhq-config-f.pdparams
[12/29 20:20:09] ppgan INFO: Found C:\Users\huyi\.cache\ppgan\stylegan2-ffhq-config-f-directions.pdparams

看看效果

效果还是不错的,还能看出是谁,对吗?

总结

PaddleGAN项目我最近开始研究,还没有很深入,如果有理解不对的地方可以留言指正。如果大家喜欢这些效果的话,我可以把其他的效果,比如性别反转等等做出来(想看的话,留言告诉我)。有时间的话也可以魔改一下代码生成点有意思的特效。

以上就是Python PaddleGAN实现调整照片人物年龄的详细内容,更多关于Python PaddleGAN调整照片人物年龄的资料请关注脚本之家其它相关文章!

相关文章

  • Python生成任意范围任意精度的随机数方法

    Python生成任意范围任意精度的随机数方法

    下面小编就为大家分享一篇Python生成任意范围任意精度的随机数方法,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧
    2018-04-04
  • python生成随机图形验证码详解

    python生成随机图形验证码详解

    这篇文章主要介绍了python生成随机图形验证码详解,具有一定参考价值,需要的朋友可以参阅。
    2017-11-11
  • Python之基础函数案例详解

    Python之基础函数案例详解

    这篇文章主要介绍了Python之基础函数案例详解,本篇文章通过简要的案例,讲解了该项技术的了解与使用,以下就是详细内容,需要的朋友可以参考下
    2021-08-08
  • 浅析Python中的序列化存储的方法

    浅析Python中的序列化存储的方法

    这篇文章主要介绍了Python中的序列化存储的方法,序列化存储主要针对的是内存和硬盘之间的写入操作,需要的朋友可以参考下
    2015-04-04
  • 基于python tornado实现图床功能

    基于python tornado实现图床功能

    因为买了阿里/腾讯的云服务器,但是使用云存储还需要收费,又加上家里正好有一台nas,又加上闲的没事,所以搞了一个小脚本,这个项目主要功能是为typora增加一个自定义图床,本文给大家介绍基于python tornado实现图床功能,感兴趣的朋友一起看看吧
    2023-08-08
  • 如何利用pygame实现简单的五子棋游戏

    如何利用pygame实现简单的五子棋游戏

    这篇文章主要给大家介绍了关于如何利用pygame实现简单的五子棋游戏的相关资料,文中通过示例代码介绍的非常详细,对大家学习或者使用pygame具有一定的参考学习价值,需要的朋友们下面来一起学习学习吧
    2019-12-12
  • python/golang实现循环链表的示例代码

    python/golang实现循环链表的示例代码

    这篇文章主要介绍了python/golang如何实现循环链表,帮助大家更好的理解和学习循环链表的实现方法,感兴趣的朋友可以了解下
    2020-09-09
  • 在Python中使用SimpleParse模块进行解析的教程

    在Python中使用SimpleParse模块进行解析的教程

    这篇文章主要介绍了在Python中使用SimpleParse模块进行解析的教程,文章来自于IBM官方的开发者技术文档,需要的朋友可以参考下
    2015-04-04
  • Python Tkinter简单布局实例教程

    Python Tkinter简单布局实例教程

    这篇文章主要介绍了Python Tkinter简单布局实例教程,包括了填充、左右布局、绝对布局、网格布局等,需要的朋友可以参考下
    2014-09-09
  • python实现自动化脚本编写

    python实现自动化脚本编写

    自动化在很多时候是很方便的,本文以修改用户名密码单元为案例,编写测试脚本。完成修改用户名密码模块单元测试,感兴趣的可以了解一下
    2021-06-06

最新评论