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研习社(okweiwu)

原文出处及转载信息见文内详细说明,如有侵权,请联系 yunjia_community@tencent.com 删除。

原始发表时间:2019-02-22

本文参与腾讯云自媒体分享计划,欢迎正在阅读的你也加入,一起分享。

发表于

我来说两句

0 条评论
登录 后参与评论

扫码关注云+社区

领取腾讯云代金券