From 9467998d18246acb60d76349826b2e8ebaedfe9c Mon Sep 17 00:00:00 2001 From: Jules Laplace Date: Sun, 17 Jun 2018 09:21:43 +0200 Subject: args --- data/recursive_dataset.py | 7 ++----- recursive.py | 1 + 2 files changed, 3 insertions(+), 5 deletions(-) diff --git a/data/recursive_dataset.py b/data/recursive_dataset.py index 14ce906..87f331c 100644 --- a/data/recursive_dataset.py +++ b/data/recursive_dataset.py @@ -11,16 +11,13 @@ class RecursiveDataset(BaseDataset): self.root = opt.dataroot ### input A (label maps) - self.dir_A = opt.dataroot - self.A_paths = sorted(make_dataset(self.dir_A)) - - self.dataset_size = len(self.A_paths) + self.dataset_size = 1000000 def __getitem__(self, index): ### input A (label maps) A_path = os.path.join(self.opt.dataroot, "frame_{:05d}.png".format(index)) if not os.path.exists(A_path): - # print() + print("{} doesn't exist, waiting for it".format(A_path)) while not os.path.exists(A_path): # print('sleeping for {}'.format(self.opt.poll_delay)) time.sleep(self.opt.poll_delay) diff --git a/recursive.py b/recursive.py index 3e24e19..61c5f62 100644 --- a/recursive.py +++ b/recursive.py @@ -44,6 +44,7 @@ start_img_path = os.path.join(render_dir, "frame_00000.png") copyfile(opt.start_img, start_img_path) for i, data in enumerate(dataset): + print("iter...{}".format(i)) if i >= opt.how_many: break if opt.data_type == 16: -- cgit v1.2.3-70-g09d2