diff options
| author | Jules Laplace <julescarbon@gmail.com> | 2018-06-17 09:21:43 +0200 |
|---|---|---|
| committer | Jules Laplace <julescarbon@gmail.com> | 2018-06-17 09:21:43 +0200 |
| commit | 9467998d18246acb60d76349826b2e8ebaedfe9c (patch) | |
| tree | b821dfca3343aa62bf32e38ea637226af6512cc3 | |
| parent | 7c437fca5f00252af4cee9541460a2c36281fbe6 (diff) | |
args
| -rw-r--r-- | data/recursive_dataset.py | 7 | ||||
| -rw-r--r-- | 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: |