导
语
人脸图像编辑广泛应用于图像后期处理中，通常使用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客栈
分享人工智能领域最新资讯，讲解各类实用的算法和模型。
原理剖析（续）
论文概述中提到了Transformer、GAN逆映射等概念，这些论文细节将在接下来的公众号文章中提到。
