diff options
Diffstat (limited to 'data/recursive_dataset.py')
| -rw-r--r-- | data/recursive_dataset.py | 12 |
1 files changed, 3 insertions, 9 deletions
diff --git a/data/recursive_dataset.py b/data/recursive_dataset.py index 828adb5..863c385 100644 --- a/data/recursive_dataset.py +++ b/data/recursive_dataset.py @@ -21,15 +21,9 @@ class RecursiveDataset(BaseDataset): A_path = os.path.join(self.opt.render_dir, "frame_{:05d}.png".format(index)) # 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... {}'.format(A_path)) - while not os.path.exists(A_path): - # print('sleeping for {}'.format(self.opt.poll_delay)) - time.sleep(self.opt.poll_delay) + while not os.path.exists(A_path): + # print('sleeping for {}'.format(self.opt.poll_delay)) + time.sleep(self.opt.poll_delay) # print("got {}".format(A_path)) A_img = Image.open(A_path).convert('RGB') A = self.transform(A_img) |