Python?PaddleGAN實現(xiàn)照片人物性別反轉

更新時間：2021年12月31日 10:53:02 作者：劍客阿良_ALiang

PaddleGAN中的styleganv2editing.py是支持性別編輯的。所以本文將介紹如何通過調整參數(shù)，來試著實現(xiàn)一下照片的性別翻轉。感興趣的小伙伴可以學習一下

前言

接著我的上篇文章：Python PaddleGAN實現(xiàn)調整照片人物年齡

在上面的文章中，我們發(fā)現(xiàn)styleganv2editing.py是支持性別編輯的。所以調整了一下參數(shù)，來試著實現(xiàn)一下照片的性別翻轉。下面我們開始吧

環(huán)境搭建

這部分就直接參考上面的文章吧，就不再寫一遍了。先發(fā)一下我準備的照片，如下：

實現(xiàn)過程

下面我們一步步操作一下，首先我們要做個預處理，和上一篇文章中一樣。

預處理

執(zhí)行命令

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

執(zhí)行結果

我們得到了與處理后的圖像

性別翻轉

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文件。

執(zhí)行結果

我們得到了性別翻轉后的圖

是不是有點男性的感覺了，但是還是覺著差點什么。

眼睛眉毛調整

我覺著可能是眼睛和眉毛的距離太短了，我們用下面的命令調整一下：

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

執(zhí)行結果

我們得到處理后的文件如下

還不錯，但是感覺還是少了點什么。

開口處理

想把嘴巴張開一點，看看感覺如何，使用下面的命令：

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

執(zhí)行結果

看一下效果

?

表情還是有點僵硬。

微笑處理

為了讓圖片笑著自然點，我們再調整一下，使用下面的代碼：

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

執(zhí)行結果