init

app.py

@@ -54,9 +54,9 @@ def run(
     paprika: ImportGraph,
 ) -> tuple[PIL.Image.Image, PIL.Image.Image, PIL.Image.Image]:
 
-    im1 = shinkai.test(image.name, True)
-    im2 = hayao.test(image.name, True)
-    im3 = paprika.test(image.name, True)
+    im1 = shinkai.test('shinkai', 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)
 

test1.py

@@ -15,7 +15,7 @@ class ImportGraph:
         with self.graph.as_default():
             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
+                self.test_generated = generator.G_net(test_real).fake
             saver = tf.train.Saver()
 
             ckpt = tf.train.get_checkpoint_state(checkpoint_dir)  # checkpoint file information
@@ -26,10 +26,13 @@ class ImportGraph:
             else:
                 print(" [*] Failed to find a checkpoint")
 
-    def test(self, sample_file, if_adjust_brightness, img_size=[256,256]):
+    def test(self, style_name, sample_file, if_adjust_brightness, img_size=[256,256]):
+        result_dir = 'results/' + style_name
+        check_folder(result_dir)
+
         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 = 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: