DeepConvolutionalGanModel: Deep convolutional GAN (DCGAN) model

In ANTsX/ANTsRNet: Neural Networks for Medical Image Processing

Deep convolutional GAN (DCGAN) model

Description

Deep convolutional generative adverserial network from the paper:

Details

https://arxiv.org/abs/1511.06434

and ported from the Keras (python) implementation:

https://github.com/eriklindernoren/Keras-GAN/blob/master/dcgan/dcgan.py

Arguments

inputImageSize

latentDimension

Details

$initialize instantiates a new class and builds the generator and discriminator. $buildGeneratorbuild generator. $buildGeneratorbuild discriminator.

Author(s)

Tustison NJ

Examples

library( keras )
library( ANTsRNet )

keras::backend()$clear_session()

# Let's use the mnist data set.

mnist <- dataset_mnist()

numberOfTrainingData <- length( mnist$train$y )

inputImageSize <- c( dim( mnist$train$x[1,,] ), 1 )

x <- array( data = mnist$train$x / 255, dim = c( numberOfTrainingData, inputImageSize ) )
y <- mnist$train$y

numberOfClusters <- length( unique( mnist$train$y ) )

# Instantiate the DCEC model

ganModel <- DeepConvolutionalGanModel$new(
   inputImageSize = inputImageSize,
   latentDimension = 100 )

ganModel$train( x, numberOfEpochs = 2 )

ANTsX/ANTsRNet documentation built on Nov. 21, 2024, 4:07 a.m.