首页 科技 正文

生成“猫狗版”特朗普,造假脸工具StarGANv2被玩坏,算法已开源

晓查 只想说 凹非寺量子位 报导 | 微信公众号 QbitAI

假如你要用AI把川普的脸变成猫,又或是是把自己的脸与好看的奥妹融合起來,你能该怎么办?

或许你能想起两种AI专用工具。可是Deepfake转化成面部违和太强,GAN转化成假脸非常难深层订制。

如今来源于日本团体开发设计的风格迁移模型StarGANv2能够极致处理这个问题

假如你是一个对GAN长期性关心的同学们,那麼一定你是否还记得StarGAN,它是日本高丽大学和Clova AI研究所在17年底发布的风格迁移模型。

只需键入一张半身照,StarGAN就能让你了解自身更改头型、性別、年纪后各自长啥样。

而StarGANv2更是同一科学研究团体发布的升級版本号,早已被CVPR 2020百度收录。近期她们在GitHub上发布了官方网完成编码,迅速就被网民玩坏了。

StarGANv2有两个训炼数据,一个对于面部,一个对于动物脸。

二者原本是“井水不犯河水”,可是许多人偏要要互用,拿川普的脸键入到动物脸预训练模型里。結果特普朗的脸转化成猫就变成那样:

与狗脸生成又变成那样,模型乃至给狗穿上了西装。

安裝应用

那么趣味的模型,如何安装呢?好在StarGANv2的开发设计团体出示了详细的编码和预训练模型。

先将新项目复制到当地:

git clone https://github.com/clovaai/stargan-v2.git

cd stargan-v2/

安裝依靠项:

conda create -n stargan-v2 python=3.6.7conda activate stargan-v2

conda install -y pytorch=1.4.0 torchvision=0.5.0 cudatoolkit=10.0 -c pytorch

conda install x264==’1!152.20180717’ ffmpeg=4.0.2 -c conda-forge

pip install opencv-python==4.1.2.30 ffmpeg-python==0.2.0 scikit-image==0.16.2

pip install pillow==7.0.0 scipy==1.2.1 tqdm==4.43.0 munch==2.5.0

随后免费下载预训炼数据,在其中前二项是人脸数据集和预训练模型,后一项是动物脸。

bash download.sh pretrained-network-celeba-hq

bash download.sh wing

bash download.sh pretrained-network-afhq

最终将必须变换的相片裁剪成和数据图片一样的宽高比,键入下列指令变换图片。

python main.py —mode sample —num_domains 2 —resume_iter 100000 —w_hpf 1 \

—checkpoint_dir expr/checkpoints/celeba_hq \

—result_dir expr/results/celeba_hq \

—src_dir assets/representative/celeba_hq/src \—ref_dir assets/representative/celeba_hq/ref

实际效果远超别的GAN

StarGANv2由四个控制模块构成:

在线生成器将键入图像变换为体现特殊域设计风格编号的輸出图像;投射互联网将潜在性代码转换为训炼中任意挑选的好几个域的设计风格编号;设计风格伺服电机获取图像的设计风格编号,容许在线生成器实行参照正确引导的图像生成;鉴别器从好几个域中区别真正图像和虚报图像。

创作者任意选择了10张图像开展风格迁移的试验,結果FID远远地低于别的类似模型。

她们还把四种模型转化成的图像上传入AMT服务平台,交到人工服务评比,绝大多数检查者都觉得StarGANv2转化成的图片更高品质。

从人工服务评比的結果还可以看得出,StarGANv2在转化成动物脸部的实际效果更强,怪不得网民要拿动物模型来更新改造特朗普头像了。

毕业论文详细地址:https://arxiv.org/abs/1912.01865v2

非特殊说明,本文由原创资讯网原创或收集发布。

转载请注明本文地址:http://www.68nw.com/kj/451.html