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

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机器上开发的

• 所提供的模型和示例代码均遵循非商业的知识共享许可协议。

接下来要做的事

• 更新训练代码

【注】本文的相关链接请扫下方二维码访问。

AI求职百题斩 · 每日一题

每天进步一点点，赶紧扫码参与每日一题！

点击下方“阅读原文”，参与 强化学习论文讨论小组 互动

↓↓↓