CVPR2022新作：P图不会，深度学习来帮忙：基于GAN逆映射的图像编辑(上)

导

语

人脸图像编辑广泛应用于图像后期处理中，通常使用PS工具通过多个步骤来完成。手动完成人像的后期编辑非常耗时，同时对工具的娴熟运用会有较高的学习门槛。下面让我们看下如何用AI一键修图。

本文效果

这篇论文全名为"Style Transformer for Image Inversion and Editing",由华东师范大学、上海统计与数据科学重点实验室、华中科技大学的多位研究员共同完成。下图为本论文实现的图像编辑效果：

1）表情转为微笑

2）性别互换

3）戴眼镜

代码实战

模型训练

python scripts/train.py \
--dataset_type=ffhq_encode \
--exp_dir=results/train_style_transformer \
--batch_size=8 \
--test_batch_size=8 \
--val_interval=5000 \
--save_interval=10000 \
--stylegan_weights=pretrained_models/stylegan2-ffhq-config-f.pt

模型推理

python scripts/inference.py \
--exp_dir=results/infer_style_transformer \

--checkpoint_path=results/train_style_transformer/   \

checkpoints/best_model.pt \

--data_path=/test_data \
--test_batch_size=8

资源下载

本论文涉及的代码和模型权重(国外网盘)，已打包放入百度网盘。关注公众号实用AI客栈，在后台回复1002即可获取下载链接。

实用AI客栈

分享人工智能领域最新资讯，讲解各类实用的算法和模型。

原理剖析（续）

论文概述中提到了Transformer、GAN逆映射等概念，这些论文细节将在接下来的公众号文章中提到。