diff options
Diffstat (limited to 'data/recursive_dataset.py')
| -rw-r--r-- | data/recursive_dataset.py | 7 |
1 files changed, 2 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) |