diff options
Diffstat (limited to 'data/recursive_dataset.py')
| -rw-r--r-- | data/recursive_dataset.py | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/data/recursive_dataset.py b/data/recursive_dataset.py index 7bf7941..1460567 100644 --- a/data/recursive_dataset.py +++ b/data/recursive_dataset.py @@ -18,12 +18,14 @@ class RecursiveDataset(BaseDataset): def __getitem__(self, index): A_path = os.path.join(self.opt.render_dir, "frame_{:05d}.png".format(index)) - # print(A_path) + print(A_path) if not os.path.exists(A_path): A_path2 = os.path.join(self.opt.render_dir, "frame_{:05d}.png".format(index+1)) if os.path.exists(A_path2): + print('oops: {}'.format(A_path2)) A_path = A_path2 else: + print('not there yet...') while not os.path.exists(A_path): gevent.sleep(self.opt.poll_delay) A_img = Image.open(A_path).convert('RGB') |