Github项目推荐 | SC-FEGAN：基于GAN的人脸照片涂鸦编辑

（使用用户的草图和颜色进行面部编辑生成对抗网络，可添加/更改耳环，眼镜，发型，酒窝等）

by Youngjoo Jo, Jongyoul Park

《SC-FEGAN: Face Editing Generative Adversarial Network with User's Sketch and Color》Y Jo, J Park [South Korea] (2019)

论文链接：

https://arxiv.org/abs/1902.06838

模型：

https://drive.google.com/drive/folders/1VPsYuIK_DY3Gw07LEjUhg2LwbEDlFpq1

Github项目地址：

https://github.com/JoYoungjoo/SC-FEGAN

【注】点击文末【阅读原文】即可访问。

概述

我们学习使用深度网络来编辑人像。我们的网络SC-FEGAN（即本项目）非常适合使用直观的用户输入与草图和颜色生成高质量的合成图像。我们使用SN-patchGAN鉴别器和Unet-like发生器和门控卷积层。

依赖项

• tensorflow
• numpy
• Python3
• PyQt5
• opencv-python
• pyyaml

使用

首先，从Google drive 下载模型

基本用法是：

mv /${HOME}/SC-FEGAN.ckpt.* /${HOME}/ckpt/
python3 demo.py

通过编辑 demo.yaml 文件选择GPU的数量（不支持多GPU）。

GPU_NUM: 1 (the number you want to use)
#GPU_NUM: (if you want to use only CPU, erase the number)

您可以使用我们的网络与简单的GUI。只填充已擦除的区域。

GUI中的按钮：

• Open Image ：打开要编辑的图像并重置草图和颜色。
• Mask ：单击此按钮并在左侧查看器上绘制蒙版。
• Sketches ：单击此按钮并在左侧查看器上绘制素描线。
• Color ：单击此按钮并绘制颜色线。 如果您第一次单击此按钮，则必须从调色板中选择颜色。
• Palette ：单击此按钮可更改颜色。 如果选择颜色，则单击“颜色”按钮进行更改。
• Save Img ：单击此按钮可保存更改后的图片。 将以'name.jpg'的方式保存。
• Arrange ：安排编辑工作。
• Undo ：撤消之前的编辑工作。
• Complete ：完成图像生成并在右侧显示。

我们建议你按照以下步骤使用：

1. 根据原图合理地画出草图。

2. 在草图区域绘制蒙版。

3. 单击“Arrange”按钮。

4. 在蒙版区域上绘制颜色。

5. 单击“Complete”。

示例

面部编辑

耳环编辑

面部修复

面部修复（只有草图和颜色）

License

CC 4.0署名 - 非商业国际

该软件仅用于教育和学术研究。

注意

• 本项目是在运行Ubuntu 18.04.1的Linux机器上开发的
• 所提供的模型和示例代码均遵循非商业的知识共享许可协议。