def main(argv):
# turn off log message
tf.compat.v1.logging.set_verbosity(tf.compat.v1.logging.FATAL)
# initializer
init_op = tf.group(tf.initializers.global_variables(),
tf.initializers.local_variables())
with tf.Session(config=utils.config(index=FLAGS.gpu_index)) as sess:
# set network
kwars = {
'sess': sess,
'noise_dim': FLAGS.noise_dim,
'image_size': FLAGS.image_size,
'generator_layer': generator_layer,
'discriminator_layer': discriminator_layer,
'is_training': False
}
Model = DCGAN(**kwars)
# initialize
sess.run(init_op)
# test
Model.restore_model(
os.path.join(FLAGS.indir, 'model',
'model_{}'.format(FLAGS.model_index)))
noise = np.random.uniform(-1.,
1.,
size=[FLAGS.batch_size, FLAGS.noise_dim])
samples = Model.gen_sample(noise)
samples = samples[:, :, :, 0]
m = utils.montage(samples)
gen_img = m
plt.axis('off')
plt.imshow(gen_img, cmap='gray')
plt.show()
