专栏首页AI研习社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-28

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

我来说两句

0 条评论
登录 后参与评论

相关文章

  • 深入理解机器学习:从原理到算法 学习笔记-第1周 01引论

      以老鼠怯饵效应为例,老鼠根据过往的经验预测所食的食物未来对自己的影响,这就是一种学习机制。再如垃圾邮件过滤机制也是如此,虽然垃圾邮件的判别可以通过已存在的邮...

    csxiaoyao
  • 情感分析的方法有哪些

    情感分析也称为意见挖掘,是自然语言处理(NLP)中的一个领域,它试图在文本中识别和提取意见

    杨熹
  • 中文NLP笔记:5. 文本数据如何转换成计算机能够计算的数据

      把文本(段落或者文档)看作无序的集合,忽略语法和单词的顺序,计算每个单词出现的次数

    杨熹
  • 主成分分析(PCA)简介

    主成分分析实例:一个平均值为(1, 3)、标准差在(0.878, 0.478)方向上为3、在其正交方向为1的高斯分布。这里以黑色显示的两个向量是这个分布的协方差...

    iOSDevLog
  • 图解什么是 Transformer

    Transformer 是 Google 团队在 17 年 6 月提出的 NLP 经典之作, 由 Ashish Vaswani 等人在 2017 年发表的论文...

    杨熹
  • 深入理解机器学习:从原理到算法 学习笔记-第1周 02简易入门

    领域集:X,例如所有木瓜的集合。 标签集:Y,目前仅讨论二元集合,如{0,1}或者{−1,+1},表示木瓜好吃和不好吃。 训练数据:形如S = ((x ...

    csxiaoyao
  • 人工智能基础

    1950 年,艾伦.图灵 (Alan Turing) 在他的论文《计算机器与智能》 ( Compu- tmg Machinery and Intelligenc...

    iOSDevLog
  • Auto ML 一种自动完成机器学习任务的系统

    在 2018 年比较火,很多大公司都开源了各自的auto ml库,例如 Cloud AutoML, AUTO KERAS, Auto Sklearn, Auto...

    杨熹
  • 机器学习模型算法 List

    Reinforcement or Semi-Supervised Machine Learning

    杨熹
  • Python3 机器学习简明教程

    1 机器学习介绍     1.1 什么是机器学习     1.2 机器学习的应用     1.3 机器学习基本流程与工作环节         1.3.1...

    iOSDevLog

扫码关注云+社区

领取腾讯云代金券