导
语
人脸图像编辑广泛应用于图像后期处理中，通常使用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
本论文涉及的代码和模型权重(国外网盘)，已打包放入百度网盘。扫码关注公众号，在后台回复1002即可获取下载链接。
实用AI客栈
分享人工智能领域最新资讯，讲解各类实用的算法和模型。
原理剖析（续）论文概述中提到了Transformer、GAN逆映射等概念，这些论文细节将在接下来的公众号文章中提到。
