欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  IT编程

Python PaddleGAN实现照片人物性别反转

程序员文章站 2022-06-24 10:40:52
前言接着我的上篇文章:python paddlegan实现调整照片人物年龄在上面的文章中,我们发现styleganv2editing.py是支持性别编辑的。所以调整了一下参数,来试着实现一下照片的性别...

前言

接着我的上篇文章:python paddlegan实现调整照片人物年龄

在上面的文章中,我们发现styleganv2editing.py是支持性别编辑的。所以调整了一下参数,来试着实现一下照片的性别翻转。下面我们开始吧

环境搭建

这部分就直接参考上面的文章吧,就不再写一遍了。先发一下我准备的照片,如下:

Python PaddleGAN实现照片人物性别反转

实现过程

下面我们一步步操作一下,首先我们要做个预处理,和上一篇文章中一样。

预处理

执行命令

python -u applications/tools/pixel2style2pixel.py --input_image d:/spyder/paddlegan/data_hy/331.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

执行结果

Python PaddleGAN实现照片人物性别反转

我们得到了与处理后的图像

Python PaddleGAN实现照片人物性别反转

性别翻转

ok,我们把照片进行一个性别翻转。使用的命令如下:

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

注意这里使用的是output里的npy文件。

执行结果

Python PaddleGAN实现照片人物性别反转

我们得到了性别翻转后的图

Python PaddleGAN实现照片人物性别反转

是不是有点男性的感觉了,但是还是觉着差点什么。

眼睛眉毛调整

我觉着可能是眼睛和眉毛的距离太短了,我们用下面的命令调整一下:

python -u applications/tools/styleganv2editing.py --latent d:/spyder/paddlegan/data_hy/gender/dst.editing.npy --output_path d:/spyder/paddlegan/data_hy/test --model_type ffhq-config-f --size 1024 --style_dim 512 --n_mlp 8 --channel_multiplier 2 --direction_name eye_eyebrow_distance --direction_offset 20 --cpu

注意用的npy文件是上一个处理完成后的文件dst.editing.npy

执行结果

Python PaddleGAN实现照片人物性别反转

我们得到处理后的文件如下

Python PaddleGAN实现照片人物性别反转

还不错,但是感觉还是少了点什么。

开口处理

想把嘴巴张开一点,看看感觉如何,使用下面的命令:

python -u applications/tools/styleganv2editing.py --latent d:/spyder/paddlegan/data_hy/test/dst.editing.npy --output_path d:/spyder/paddlegan/data_hy/test1 --model_type ffhq-config-f --size 1024 --style_dim 512 --n_mlp 8 --channel_multiplier 2 --direction_name mouth_open --direction_offset 20 --cpu

执行结果

Python PaddleGAN实现照片人物性别反转

看一下效果

Python PaddleGAN实现照片人物性别反转 

表情还是有点僵硬。

微笑处理

为了让图片笑着自然点,我们再调整一下,使用下面的代码:

python -u applications/tools/styleganv2editing.py --latent d:/spyder/paddlegan/data_hy/test1/dst.editing.npy --output_path d:/spyder/paddlegan/data_hy/test2 --model_type ffhq-config-f --size 1024 --style_dim 512 --n_mlp 8 --channel_multiplier 2 --direction_name smile --direction_offset 2 --cpu

执行结果

Python PaddleGAN实现照片人物性别反转

看一下最终的效果。

Python PaddleGAN实现照片人物性别反转

是不是效果很不错。我们放一下对比效果。

Python PaddleGAN实现照片人物性别反转

总体效果还是很理想的。

总结

本文是对上面这篇文章的补充使用,因为支持编辑,所以可以按照需求对图像做调整。如果大家对这个项目感兴趣,不妨拉下来自己玩玩看,各种奇思妙想的弄一弄哦。

到此这篇关于python paddlegan实现照片人物性别反转的文章就介绍到这了,更多相关python paddlegan照片人物性别反转内容请搜索以前的文章或继续浏览下面的相关文章希望大家以后多多支持!