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 ++----- 1 file changed, 2 insertions(+), 5 deletions(-) (limited to 'data/recursive_dataset.py') 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) -- cgit v1.2.3-70-g09d2