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

2019 年 2 月 22 日 AI研习社

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求职百题斩 · 每日一题


每天进步一点点,赶紧扫码参与每日一题!




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

↓↓↓ 
登录查看更多
13

相关内容

SC:International Conference for High Performance Computing, Networking, Storage, and Analysis。 Explanation:高性能计算、网络、存储和分析国际会议。 Publisher:IEEE。 SIT: http://dblp.uni-trier.de/db/conf/sc/
专知会员服务
109+阅读 · 2020年3月12日
必读的10篇 CVPR 2019【生成对抗网络】相关论文和代码
专知会员服务
31+阅读 · 2020年1月10日
【GitHub实战】Pytorch实现的小样本逼真的视频到视频转换
专知会员服务
35+阅读 · 2019年12月15日
【ICIP2019教程-NVIDIA】图像到图像转换,附7份PPT下载
专知会员服务
53+阅读 · 2019年11月20日
Keras作者François Chollet推荐的开源图像搜索引擎项目Sis
专知会员服务
29+阅读 · 2019年10月17日
Github 项目推荐 | PyTorch 实现的 GAN 文本生成框架
AI研习社
33+阅读 · 2019年6月10日
Github项目推荐 | Pytorch TVM 扩展
AI研习社
11+阅读 · 2019年5月5日
项目 | 基于GAN的人脸照片涂鸦编辑
机器学习算法与Python学习
5+阅读 · 2019年3月1日
Github 项目推荐 | 用 PyTorch 0.4 实现的 YoloV3
AI研习社
9+阅读 · 2018年8月11日
【学界】实景照片秒变新海诚风格漫画:清华大学提出CartoonGAN
GAN生成式对抗网络
14+阅读 · 2018年6月20日
CycleGAN:图片风格,想换就换 | ICCV 2017论文解读
PaperWeekly
12+阅读 · 2018年3月14日
StarGAN-多领域图像翻译
CreateAMind
4+阅读 · 2017年12月26日
Revisiting CycleGAN for semi-supervised segmentation
Arxiv
3+阅读 · 2019年8月30日
Interpretable Adversarial Training for Text
Arxiv
5+阅读 · 2019年5月30日
Arxiv
7+阅读 · 2018年12月10日
VIP会员
相关资讯
Github 项目推荐 | PyTorch 实现的 GAN 文本生成框架
AI研习社
33+阅读 · 2019年6月10日
Github项目推荐 | Pytorch TVM 扩展
AI研习社
11+阅读 · 2019年5月5日
项目 | 基于GAN的人脸照片涂鸦编辑
机器学习算法与Python学习
5+阅读 · 2019年3月1日
Github 项目推荐 | 用 PyTorch 0.4 实现的 YoloV3
AI研习社
9+阅读 · 2018年8月11日
【学界】实景照片秒变新海诚风格漫画:清华大学提出CartoonGAN
GAN生成式对抗网络
14+阅读 · 2018年6月20日
CycleGAN:图片风格,想换就换 | ICCV 2017论文解读
PaperWeekly
12+阅读 · 2018年3月14日
StarGAN-多领域图像翻译
CreateAMind
4+阅读 · 2017年12月26日
Top
微信扫码咨询专知VIP会员