Skip to content

Boundary Equilibrium Generative Adversarial Networks (BEGAN) Implementation in Microsoft Cognitive Toolkit (CNTK)

Notifications You must be signed in to change notification settings

2wins/BEGAN-cntk

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

2 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

BEGAN in CNTK

Microsoft Cognitive Toolkit (CNTK) implementation of BEGAN: Boundary Equilibrium Generative Adversarial Networks

Prerequisites

Note: I found that lower versions of CNTK caused abnormal training results of the GAN.

Usage

First, download MNIST dataset:

$ python install_mnist.py

Second, train the GAN:

$ python train.py

Third, generate digits with the trained generator:

$ python test.py --num_imgs=1000 --checkpointPath=models/BEGAN_G_274100.dnn

Result on MNIST

From scratch to 300k (frames captured every 1000 iter.)

gamma=1.0

Generated images with a trained model

About

Boundary Equilibrium Generative Adversarial Networks (BEGAN) Implementation in Microsoft Cognitive Toolkit (CNTK)

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages