From 6caef752e3be24037024c83ea3c2f4d3de3c7f37 Mon Sep 17 00:00:00 2001 From: Jules Laplace Date: Tue, 15 May 2018 02:47:42 +0200 Subject: test --- data/recursive_dataset.py | 3 ++- test-mogrify.py | 2 +- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/data/recursive_dataset.py b/data/recursive_dataset.py index 5cc3e4a..b1e0407 100644 --- a/data/recursive_dataset.py +++ b/data/recursive_dataset.py @@ -20,8 +20,9 @@ class RecursiveDataset(BaseDataset): self.transform = get_transform(opt) def __getitem__(self, index): - A_path = self.opt.results_dir + self.opt.tag + "/frame_{:05d}.png".format(index) + A_path = self.opt.render_dir + self.opt.tag + "/frame_{:05d}.png".format(index) while not os.path.exists(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 2f2203f..83b9f81 100644 --- a/test-mogrify.py +++ b/test-mogrify.py @@ -37,7 +37,7 @@ if __name__ == '__main__': opt.no_flip = True # no flip opt.experiment = data_opt.experiment # opt.start_img.split("/")[-1].split(".")[0] - render_dir = opt.results_dir + opt.name + "/" + data_opt.experiment + "/" + opt.render_dir = render_dir = opt.results_dir + opt.name + "/" + data_opt.experiment + "/" print("create render_dir: {}".format(render_dir)) if os.path.exists(render_dir): -- cgit v1.2.3-70-g09d2