Flux.jl-DCGAN

DCGAN implementation with Flux.jl

model-zoo includes a DCGAN model based on this: https://github.com/FluxML/model-zoo

I no longer maintain this repository.

Environment

Google Cloud Platform, n1-standard-8 + 1 x NVIDIA Tesla K80
Docker version 19.03.4, build 9013bf583a
Julia 1.3.0, Flux 0.10.0

(v1.3) pkg> st
    Status `~/.julia/environments/v1.3/Project.toml`
  [fbb218c0] BSON v0.2.4
  [31c24e10] Distributions v0.21.9
  [5789e2e9] FileIO v1.1.0
  [587475ba] Flux v0.10.0
  [f67ccb44] HDF5 v0.12.5
  [6218d12a] ImageMagick v0.7.5
  [916415d5] Images v0.18.0
  [4138dd39] JLD v0.9.1

Run

sudo docker build -t matsueushi/flux . 
sudo docker run --name flux --gpus all -it -v $PWD:/tmp -w /tmp matsueushi/flux:latest /bin/bash
julia src/mnist-dcgan.jl

Result

30 epochs (14,000 iterations)

References

• Generative Adversarial Networks
• GANについて概念から実装まで　～DCGANによるキルミーベイベー生成～
• はじめてのGAN
• GANによる二次元美少女画像生成
• How to Train a GAN? Tips and tricks to make GANs work

Flux implementation

• Added Condtional GAN and DCGAN tutorial #111

Implementations with other frameworks

• Tensorflow tutorial
• znxlwm/tensorflow-MNIST-GAN-DCGAN
• PyTorch tutorial
• jacobgil/keras-dcgan