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

Z时代
2024-01-10
分类：综合

前言

接着我的上篇文章：Python PaddleGAN实现调整照片人物年龄

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

环境搭建

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

实现过程

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

预处理

执行命令

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

执行结果

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

性别翻转

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 -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 -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 -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

执行结果