• Stars
    star
    222
  • Rank 179,123 (Top 4 %)
  • Language
    Python
  • License
    MIT License
  • Created over 6 years ago
  • Updated over 2 years ago

Reviews

There are no reviews yet. Be the first to send feedback to the community and the maintainers!

Repository Details

Implementation of some different variants of GANs by tensorflow, Train the GAN in Google Cloud Colab, DCGAN, WGAN, WGAN-GP, LSGAN, SNGAN, RSGAN, RaSGAN, BEGAN, ACGAN, PGGAN, pix2pix, BigGAN

DCGAN_LSGAN_WGAN_WGAN-GP_SNGAN_RSGAN_BEGAN_ACGAN_PGGAN_pix2pix_BigGAN

Implementation of some different variants of GANs

Introduction


This code is mainly implement some basic GANs about 'DCGAN', 'WGAN', 'WGAN-GP', 'LSGAN', 'SNGAN', 'RSGAN'&'RaSGAN', 'BEGAN', 'ACGAN', 'PGGAN', 'pix2pix', 'BigGAN'.

More details of these GANs, please see follow papers:

  1. DCGAN: Unsupervised representation learning with deep convolutional generative adversarial networks

  2. WGAN: Wasserstein gan

  3. WGAN-GP: Improved training of wasserstein gans

  4. LSGAN: Least Squares Generative Adversarial Networks

  5. SNGAN: Spectral normalization for generative adversarial networks

  6. RSGAN&RaSGAN: The relativistic discriminator: a key element missing from standard GAN

  7. BEGAN:BEGAN: Boundary Equilibrium Generative Adversarial Networks

  8. ACGAN: Conditional Image Synthesis With Auxiliary Classifier GANs

  9. PGGAN: Progressive Growing of GANs for Improved Quality, Stability, and Variation

  10. pix2pix: Image-to-Image Translation with Conditional Adversarial Networks

  11. BigGAN: Large Scale GAN Training for High Fidelity Natural Image Synthesis [Code]

Attention

If your computer don't have GPU to accelerate the training process, please click Google Cloud Colab to train the GANs.

How to use

Firstly, you should download the data 'facedata.mat' from Baidu Drive or Google Drive, then put the file 'facedata.mat' into the folder 'TrainingSet'.

Requirements

  1. python3.5
  2. tensorflow1.4.0
  3. pillow
  4. scipy
  5. numpy

Results of this code

This result is using DCGAN trained about 8000 iterations.

Compare LSGAN, WGAN, WGAN-GP, SNGAN, RSGAN of different iteration

Convergence of BEGAN

ACGAN for face generating

dataset: download address: Baidu Drive password: 5egd

Fixed label, change noise slightly Fixed noise, change label slightly

PGGAN for face generating

SNGAN for cifar-10

D_loss G_loss results

Pix2Pix

Dataset: Google maps download address: http://efrosgans.eecs.berkeley.edu/pix2pix/datasets/maps.tar.gz

Edges2Shoes download address: http://efrosgans.eecs.berkeley.edu/pix2pix/datasets/edges2shoes.tar.gz

More Repositories

1

PatchMatch

The algorithm of patch match implemented by python
Python
248
star
2

DragGAN_pytorch

Unofficial implementation of DragGAN with StyleGAN2/3 pretrained models
Python
93
star
3

Chinese-Character-and-Calligraphic-Image-Processing

Some interesting method like style transfer, GAN, deep neural networks for Chinese character and calligraphic image processing
Python
87
star
4

DnCNN-Denoise-Gaussian-noise-TensorFlow

Simple implementation of the paper (DnCNN)'Beyond a Gaussian Denoiser: Residual Learning of Deep CNN for Image Denoising'
Python
60
star
5

Calligraphic-Images-Denoising-by-GAN

Calligraphic Images Denoising by GAN
Python
57
star
6

Face-Attribute-Editing-StyleGAN3

Face editing by e4e, text2stylegan,interfacegan,ganspace
Python
50
star
7

Style-transfer-with-neural-algorithm

Implementation of style transfer by tensorflow, for detail please see the paper "Image Style Transfer Using Convolutional Neural Networks"(CVPR2016)
Python
50
star
8

StyleSwap

Unofficial implementation of the paper: StyleSwap: Style-Based Generator Empowers Robust Face Swapping
Python
50
star
9

sngan_projection_TensorFlow

Simply implement the paper: cGANs with Projection Discriminator by TensorFlow
Python
40
star
10

Conditional-Instance-Norm-for-n-Style-Transfer

Implementation of the paper A Learned Representation for Artistic Style (Conditional instance normalization)
Python
40
star
11

CNN-for-Chinese-Calligraphy-Styles-classification

A convolution neural network with SE block and haar wavelet block for Chinese calligraphy styles classification by TensorFlow.(Paper: A novel CNN structure for fine-grained classification of Chinesecalligraphy styles)
Python
38
star
12

AnimateAnyone_unofficial

Unofficial implementation of Animate Anyone: Consistent and Controllable Image-to-Video Synthesis for Character Animation
Python
36
star
13

Learning-Residual-Images-for-Face-Attribute-Manipulation

The code of the paper 'Learning-Residual-Images-for-Face-Attribute-Manipulation' is implemented by TensorFlow
Python
35
star
14

Deep-image-analogy-TensorFlow

Implementation of the paper : Deep image analogy
Python
34
star
15

BigGAN-tensorflow

Reimplementation of the Paper: Large Scale GAN Training for High Fidelity Natural Image Synthesis
Python
28
star
16

Deep-Feature-Interporlation-Face-Attribute-manipulation-Glasses-Remove-Youth2Senior-etc.-TensorFlow

A simple implementation of the paper 'Deep Feature Interpolation for Image Content Changes'
Python
28
star
17

SRGAN-with-WGAN-Loss-TensorFlow

SRGAN (super resolution generative adversarial networks) with WGAN loss function in TensorFlow
Python
26
star
18

Residual-Dense-Network-Trained-with-cGAN-for-Super-Resolution

This repository is as a research project in the field of super resolution. It uses RDN as the generator and spectral norm is used in discriminator.
Python
25
star
19

Summary-of-deep-learning-papers

Collecting some interesting papers about computer vision
21
star
20

CycleGAN

A simple code of CycleGAN which is easy to read is implemented by TensorFlow
Python
20
star
21

DDPM_pytorch

Denoising Diffusion Probabilistic Models (DDPM)
Python
20
star
22

Semantic-Image-Inpainting-face-inpainting-TensorFlow

Simplely implement the paper 'Semantic Image Inpainting with Deep Generative Models' by TensorFlow
Python
19
star
23

FormResNet-Denoise-Gaussian-noise-TensorFlow

Simplely implement the paper 'FormResNet: Formatted Residual Learning for Image Restoration' by TensorFlow
Python
17
star
24

Real-Time-Arbitrary-Style-Transfer-AdaIN-TensorFlow

This code mainly implement the paper ' Arbitrary Style Transfer in Real-time with Adaptive Instance Normalization ' by TensorFlow
Python
17
star
25

RetinaNet_TensorFlow

Implementation of RetinaNet (focal loss) by TensorFlow (object detection)
Python
16
star
26

CartoonGAN-tensorflow

Simple code implement the paper of CartoonGAN(CVPR2018)
Python
16
star
27

yolo_v1_v2_tensorflow

Simple implementation of yolo v1 and yolo v2 by TensorFlow
Python
15
star
28

CartoonBANK

We are committed to building an cartoon community using deep generative models
Python
13
star
29

face-guided-diffusion

Unofficial implementation guided diffusion trained with Celeba
Python
13
star
30

ContextEncoder_Cat-s_head_Inpainting_TensorFlow

Inplementing the paper 'Context Encoders: Feature Learning by Inpainting' by TensorFlow
Python
12
star
31

CapsuleNet_Tensorflow

The code of CapsuleNet
Python
11
star
32

RetinaNet_BiFPN_TensorFlow

Simply implement the RetinaNet with BiFPN (EfficientDet) by tensorflow
Python
11
star
33

Char-CNN-for-Text-Classification

Character-level Convolutional Networks for Text Classification
Python
8
star
34

RNN-TensorFlow

Some interesting applications of RNN, e.g. char rnn (pomes generation), seq2seq (machine translation), image captioning (NIC)
Python
8
star
35

DenseNet-TensorFlow

DenseNet for cifar10
Python
7
star
36

few-shot-neural-talking-head-tensorflow

Few-Shot Adversarial Learning of Realistic Neural Talking Head Models TensorFlow
Python
6
star
37

Faster_RCNN_TensorFlow

Simple implementation of faster rcnn by TensorFlow, containing 4 step training and easy to understand
Python
6
star
38

Simple-Classifier-by-back-propatation

Implementing a simple classifier by numpy
Python
6
star
39

SinGAN_Pytorch

Reimplementing the paper "SinGAN: Learning a Generative Model from a Single Natural Image"
Python
5
star
40

JoJoGAN

Unofficial implementation of the paper: JoJoGAN: One Shot Face Stylization
Python
4
star
41

Pytorch_Learning_Diary

Learning pytorch on TensorFlow basis, and write down every error I meet.
Python
3
star
42

FCN_Segmentation_TensorFlow

Implementing the paper: Fully Convolutional Networks for Semantic Segmentation
Python
3
star