INNER CODE UNIT · Python
build_sn_patch_gan_discriminator
zuruoke/watermark-removal · inpaint_model.py:112
def build_sn_patch_gan_discriminator(self, x, reuse=False, training=True):
with tf.variable_scope('sn_patch_gan', reuse=reuse):
cnum = 64
x = dis_conv(x, cnum, name='conv1', training=training)
x = dis_conv(x, cnum*2, name='conv2', training=training)
x = dis_conv(x, cnum*4, name='conv3', training=training)
x = dis_conv(x, cnum*4, name='conv4', training=training)
x = dis_conv(x, cnum*4, name='conv5', training=training)
x = dis_conv(x, cnum*4, name='conv6', training=training)
x = flatten(x, name='flatten')
return x
def build_gan_discriminator(
self, batch, reuse=False, training=True):
with tf.variable_scope('discriminator', reuse=reuse):
d = self.build_sn_patch_gan_discriminator(
batch, reuse=reuse, training=training)
return d