init

app.py

@@ -55,10 +55,10 @@ def run(
 ) -> tuple[PIL.Image.Image, PIL.Image.Image, PIL.Image.Image]:
 
     im1 = shinkai.test('shinkai', image.name, True)
-    im2 = hayao.test('hayao', image.name, True)
-    im3 = paprika.test('paprika', image.name, True)
+    #im2 = hayao.test('hayao', image.name, True)
+    #im3 = paprika.test('paprika', image.name, True)
 
-    return PIL.Image.open(im1),PIL.Image.open(im2),PIL.Image.open(im3)
+    return PIL.Image.open(im1),PIL.Image.open(im1),PIL.Image.open(im1)
 
 
 def main():
@@ -69,10 +69,10 @@ def main():
     curPath = os.path.abspath(os.path.dirname(__file__))
     #init
     shinkai = ImportGraph(checkpoint_dir=os.path.join(curPath,'animeganv2/checkpoint/generator_Shinkai_weight'))
-    hayao = ImportGraph(checkpoint_dir=os.path.join(curPath,'animeganv2/checkpoint/generator_Hayao_weight'))
-    paprika = ImportGraph(checkpoint_dir=os.path.join(curPath,'animeganv2/checkpoint/generator_Paprika_weight'))
+    #hayao = ImportGraph(checkpoint_dir=os.path.join(curPath,'animeganv2/checkpoint/generator_Hayao_weight'))
+    #paprika = ImportGraph(checkpoint_dir=os.path.join(curPath,'animeganv2/checkpoint/generator_Paprika_weight'))
 
-    func = functools.partial(run, shinkai=shinkai,hayao=hayao,paprika=paprika  )
+    func = functools.partial(run, shinkai=shinkai )
     func = functools.update_wrapper(func, run)
 
     

test1.py

@@ -13,12 +13,17 @@ 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():
-            saver = tf.train.Saver()
+
+            test_real = tf.placeholder(tf.float32, [1, None, None, 3], name='test')
+            with tf.variable_scope("generator", reuse=False):
+                self.test_generated = generator.G_net(test_real).fake
+
+            self.saver = tf.train.Saver()
 
             ckpt = tf.train.get_checkpoint_state(checkpoint_dir)  # checkpoint file information
             if ckpt and ckpt.model_checkpoint_path:
                 ckpt_name = os.path.basename(ckpt.model_checkpoint_path)  # first line
-                saver.restore(self.sess, os.path.join(checkpoint_dir, ckpt_name))
+                self.saver.restore(self.sess, os.path.join(checkpoint_dir, ckpt_name))
                 print(" [*] Success to read {}".format(os.path.join(checkpoint_dir, ckpt_name)))
             else:
                 print(" [*] Failed to find a checkpoint")
@@ -30,11 +35,7 @@ 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)))
 
-        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})
+        fake_img = self.sess.run(self.test_generated, feed_dict={test_real: sample_image})
         if if_adjust_brightness:
             save_images(fake_img, image_path, sample_file)
         else: