test

2 files changed, 7 insertions, 10 deletions

diff --git a/data/recursive_dataset.py b/data/recursive_dataset.py
index 0605458..b14bd66 100644
--- a/data/recursive_dataset.py
+++ b/data/recursive_dataset.py
@@ -22,7 +22,7 @@ class RecursiveDataset(BaseDataset):
     def __getitem__(self, index):
         A_path = self.opt.render_dir + "/frame_{:05d}.png".format(index)
         while not os.path.exists(A_path):
-            print(A_path)
+            # print(A_path)
             time.sleep(0.01)
         A_img = Image.open(A_path).convert('RGB')
         A = self.transform(A_img)
diff --git a/test-mogrify.py b/test-mogrify.py
index 83b9f81..2ffadbb 100644
--- a/test-mogrify.py
+++ b/test-mogrify.py
@@ -21,23 +21,20 @@ from time import sleep
 if __name__ == '__main__':
     opt = TestOptions().parse()
     data_opt = DatasetOptions().parse(opt.unknown)
+    opt.nThreads = 1   # test code only supports nThreads = 1
+    opt.batchSize = 1  # test code only supports batchSize = 1
+    opt.serial_batches = True  # no shuffle
+    opt.no_flip = True  # no flip
+    opt.experiment = data_opt.experiment # opt.start_img.split("/")[-1].split(".")[0]
 
     d = datetime.now()
     tag = "{}_{}_{}_{}".format(
         opt.name, opt.experiment, data_opt.experiment,
-        # opt.how_many, frac_a,
-        # blur, sigma, canny_lo, canny_hi,
         d.strftime('%Y%m%d%H%M'))
 
     opt.tag = tag
 
-    opt.nThreads = 1   # test code only supports nThreads = 1
-    opt.batchSize = 1  # test code only supports batchSize = 1
-    opt.serial_batches = True  # no shuffle
-    opt.no_flip = True  # no flip
-    opt.experiment = data_opt.experiment # opt.start_img.split("/")[-1].split(".")[0]
-
-    opt.render_dir = render_dir = opt.results_dir + opt.name + "/" + data_opt.experiment + "/"
+    opt.render_dir = render_dir = opt.results_dir + opt.name + "/" + tag + "/"
 
     print("create render_dir: {}".format(render_dir))
     if os.path.exists(render_dir):