MCPcopy Index your code
hub / github.com/tensorpack/tensorpack / generator

Method generator

examples/GAN/DCGAN.py:46–58  ·  view source on GitHub ↗

return an image generated from z

(self, z)

Source from the content-addressed store, hash-verified

44 return [tf.TensorSpec((None, self.shape, self.shape, 3), tf.float32, 'input')]
45
46 def generator(self, z):
47 """ return an image generated from z"""
48 nf = 64
49 l = FullyConnected('fc0', z, nf * 8 * 4 * 4, activation=tf.identity)
50 l = tf.reshape(l, [-1, 4, 4, nf * 8])
51 l = BNReLU(l)
52 with argscope(Conv2DTranspose, activation=BNReLU, kernel_size=4, strides=2):
53 l = Conv2DTranspose('deconv1', l, nf * 4)
54 l = Conv2DTranspose('deconv2', l, nf * 2)
55 l = Conv2DTranspose('deconv3', l, nf)
56 l = Conv2DTranspose('deconv4', l, 3, activation=tf.identity)
57 l = tf.tanh(l, name='gen')
58 return l
59
60 @auto_reuse_variable_scope
61 def discriminator(self, imgs):

Callers 2

build_graphMethod · 0.95
build_graphMethod · 0.95

Calls 4

FullyConnectedFunction · 0.85
BNReLUFunction · 0.85
argscopeFunction · 0.85
Conv2DTransposeFunction · 0.85

Tested by

no test coverage detected