init

test1.py

@@ -13,9 +13,6 @@ class ImportGraph:
         self.graph = tf.Graph()
         self.sess = tf.Session(graph=self.graph, config=tf.ConfigProto(allow_soft_placement=True, gpu_options=tf.GPUOptions(allow_growth=True)))
         with self.graph.as_default():
-            test_real = tf.placeholder(tf.float32, [1, None, None, 3], name='test')
-            with tf.variable_scope("generator", reuse=True):
-                self.test_generated = generator.G_net(test_real).fake
             saver = tf.train.Saver()
 
             ckpt = tf.train.get_checkpoint_state(checkpoint_dir)  # checkpoint file information
@@ -32,7 +29,12 @@ class ImportGraph:
 
         sample_image = np.asarray(load_test_data(sample_file, img_size))
         image_path = os.path.join(result_dir, '{0}'.format(os.path.basename(sample_file)))
-        fake_img = self.sess.run(self.test_generated, feed_dict={test_real: sample_image})
+
+        test_real = tf.placeholder(tf.float32, [1, None, None, 3], name='test')
+        with tf.variable_scope("generator", reuse=False):
+            test_generated = generator.G_net(test_real).fake
+
+        fake_img = self.sess.run(test_generated, feed_dict={test_real: sample_image})
         if if_adjust_brightness:
             save_images(fake_img, image_path, sample_file)
         else: