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()