StyleGAN - 基于 TensorFlow 的人脸图像生成器


CC BY-NC 4.0
跨平台
Python

软件简介

StyleGAN 是官方的 TensorFlow 实现,用于生成人脸图像。

这些人不是真实的 - 他们由生成器生成

该库基于论文《用于生成对抗网络的基于样式的生成器架构》( A Style-Based Generator Architecture for
Generative Adversarial Networks
)进行实现:

论文摘要:我们提出了一种用于生成对抗网络的替代生成器架构,借鉴了样式转移文献。新的架构导致高级属性的自动学习,无监督分离(例如,在人脸上训练时的姿势和身份)和生成的图像(例如,雀斑,头发)中的随机变化,它可以实现对合成的直观、特定尺度控制。新的生成器在传统的分布质量指标方面采用了最先进的技术,形成明显更好的插值特性,并且更好地解开了潜在的变异因素。为了量化插值质量,我们提出了两种新的自动化方法。最后,我们介绍了一个新的、高度多样化和高质量的人脸数据集。

系统要求

  • Linux 和 Windows 都受支持,但出于性能和兼容性的原因,强烈建议使用 Linux。
  • 64 位 Python 3.6 。推荐使用 numpy 1.14.3 或更新的 Anaconda3。
  • TensorFlow 1.10.0 或更高版本,支持 GPU。
  • 一个或多个具有至少 11GB DRAM 的高端 NVIDIA GPU。我们推荐配备 8 个 Tesla V100 GPU 的 NVIDIA DGX-1。
  • NVIDIA 驱动程序 391.35 或更高版本,CUDA 工具包 9.0 或更高版本,cuDNN 7.3.1 或更高版本。